Spaces:

koti-malla
/

object_detection

Running

koti-malla commited on Oct 7, 2023

Commit

0fd272d

•

1 Parent(s): d4e1493

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def index():
     if request.method == 'POST':
         # Load the YOLOv8 model
-        yolo_model = YOLO('yolo/yolov8s.pt')
         # Load the DETR model
         processor = DetrImageProcessor.from_pretrained("detr")

     if request.method == 'POST':
         # Load the YOLOv8 model
+        yolo_model = YOLO('yolo/best.pt')
         # Load the DETR model
         processor = DetrImageProcessor.from_pretrained("detr")