Update app.py

app.py

@@ -224,10 +224,25 @@ def create_mask(image, x1, y1, x2, y2):
 
 
 def special_process_image_yolo(risk_level, image_path, point1, point2, thresholds=None):
+    import os
+    import torch
+    from ultralytics import YOLO
+    
+    # デバイスの確認
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
-    print(device)
-    # YOLOv8モデルをロードし、GPUに移動
-    model = YOLO('./1026.pt')  # モデルのパスを指定
+    print(f"Using device: {device}")
+    
+    # モデルファイルのパス
+    model_path = './1026.pt'
+    
+    # モデルファイルの存在確認
+    if not os.path.isfile(model_path):
+        raise FileNotFoundError(f"モデルファイル '{model_path}' が見つかりません。パスを確認してください。")
+    else:
+        # YOLOv8モデルをロードし、指定デバイスに移動
+        model = YOLO(model_path).to(device)
+        print("モデルが正常にロードされ、デバイスに移動しました。")
+
     model.to(device)  # モデルをGPUに移動