Spaces:

sumit-ai-ml
/

SAM-with-YOLO

Runtime error

sumit-ai-ml commited on Oct 3, 2023

Commit

da2fab4

1 Parent(s): dff6be9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,9 @@ from ultralytics import YOLO
 # Define available YOLO models
 available_models = {
-    "Model 1 (best.pt)": YOLO("best.pt"),
-    "Model 2 (another_model.pt)": YOLO("another_model.pt"),
     # Add more models as needed
 }

 # Define available YOLO models
 available_models = {
+    "X-ray": YOLO("xray.pt"),
+    "CT scan": YOLO("CT.pt"),
+    "Ultrasound": YOLO("ultrasound.pt"),
     # Add more models as needed
 }