Spaces:

sumit-ai-ml
/

SAM-with-YOLO

Runtime error

sumit-ai-ml commited on Oct 3, 2023

Commit

425f058

1 Parent(s): 3284380

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,7 +4,7 @@ import cv2
 from PIL import Image
 from ultralytics import YOLO
 import torch
 # Define available YOLO models
 available_models = {
     "X-ray": YOLO("xray.pt"),

 from PIL import Image
 from ultralytics import YOLO
 import torch
+from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor
 # Define available YOLO models
 available_models = {
     "X-ray": YOLO("xray.pt"),