Update app.py

app.py

@@ -102,7 +102,10 @@ def process_image():
     # Check contents in the root directory
     print("Current directory contents:", os.listdir('/'))
 
-    model = YOLOv10(f'./runs/detect/train3/weights/best (1).pt')
+    model = YOLOv10(f'./runs/detect/train3/weights/best.pt')
+    device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+    model.to(device)
+
     dataset = sv.DetectionDataset.from_yolo(
         images_directory_path=f"./data/MyNewVersion5.0Dataset/valid/images",
         annotations_directory_path=f"./data/MyNewVersion5.0Dataset/valid/labels",