Spaces:

vaugheu
/

yoloEYE

Running

vaugheu commited on Jun 24

Commit

b0a2960

•

1 Parent(s): d1c20d5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ options = st.sidebar.radio(
 # YOLOv8 Model
 if model_type == 'YOLOv8':
-    path_model_file = 'yolov8.pt'
     from ultralytics import YOLO
     model = YOLO(path_model_file)

 # YOLOv8 Model
 if model_type == 'YOLOv8':
+    path_model_file = 'yolov8m.pt'
     from ultralytics import YOLO
     model = YOLO(path_model_file)