rein0421 commited on
Commit
c410ece
·
verified ·
1 Parent(s): 2a85226

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -227,7 +227,7 @@ def special_process_image_yolo(risk_level, image_path, point1, point2, threshold
227
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
228
  print(device)
229
  # YOLOv8モデルをロードし、GPUに移動
230
- model = YOLO('/app/1026.pt') # モデルのパスを指定
231
  model.to(device) # モデルをGPUに移動
232
 
233
 
 
227
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
228
  print(device)
229
  # YOLOv8モデルをロードし、GPUに移動
230
+ model = YOLO('./1026.pt') # モデルのパスを指定
231
  model.to(device) # モデルをGPUに移動
232
 
233