Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
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 |
|