rein0421 commited on
Commit
e93931a
·
verified ·
1 Parent(s): b2b7031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -239,9 +239,10 @@ def special_process_image_yolo(risk_level, image_path, point1, point2, threshold
239
  if not os.path.isfile(model_path):
240
  raise FileNotFoundError(f"モデルファイル '{model_path}' が見つかりません。パスを確認してください。")
241
  else:
 
242
  # YOLOv8モデルをロードし、指定デバイスに移動
243
  model = YOLO(model_path).to(device)
244
- print("モデルが正常にロードされ、デバイスに移動しました。")
245
 
246
  model.to(device) # モデルをGPUに移動
247
 
 
239
  if not os.path.isfile(model_path):
240
  raise FileNotFoundError(f"モデルファイル '{model_path}' が見つかりません。パスを確認してください。")
241
  else:
242
+ print("モデルが正常にロードされ、デバイスに移動しました。")
243
  # YOLOv8モデルをロードし、指定デバイスに移動
244
  model = YOLO(model_path).to(device)
245
+
246
 
247
  model.to(device) # モデルをGPUに移動
248