CloudAnts commited on
Commit
eed0fda
·
verified ·
1 Parent(s): 08c8041

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -103,8 +103,8 @@ def process_image():
103
  print("Current directory contents:", os.listdir('/'))
104
 
105
  model = YOLOv10(f'./runs/detect/train3/weights/best.pt')
106
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
107
- model.to(device)
108
 
109
  dataset = sv.DetectionDataset.from_yolo(
110
  images_directory_path=f"./data/MyNewVersion5.0Dataset/valid/images",
 
103
  print("Current directory contents:", os.listdir('/'))
104
 
105
  model = YOLOv10(f'./runs/detect/train3/weights/best.pt')
106
+ #device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
107
+ #model.to(device)
108
 
109
  dataset = sv.DetectionDataset.from_yolo(
110
  images_directory_path=f"./data/MyNewVersion5.0Dataset/valid/images",