Spaces:

banao-tech
/

omniapi

Sleeping

banao-tech commited on Feb 4

Commit

3d30079

verified ·

1 Parent(s): 85b9952

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -29,7 +29,7 @@ from utils import get_yolo_model
 import torch
 # Load YOLO model using official method
-yolo_model = get_yolo_model(model_path="weights/icon_detect/best.pt")
 # Handle device placement
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 import torch
 # Load YOLO model using official method
+yolo_model = get_yolo_model(model_path="weights/best.pt")
 # Handle device placement
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")