CloudAnts commited on
Commit
b967d8f
·
verified ·
1 Parent(s): 5a5d448

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def process_image():
102
  # Check contents in the root directory
103
  print("Current directory contents:", os.listdir('/'))
104
 
105
- model = YOLOv10(f'./runs/detect/train3/weights/best.pt', weights_only=False)
106
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
107
  model.to(device)
108
 
 
102
  # Check contents in the root directory
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