Update app.py

app.py

@@ -7,8 +7,13 @@ from PIL import Image, ImageDraw
 import numpy as np
 from sklearn.metrics import precision_recall_fscore_support
 
-# 載入 YOLOv8 模型
-yolo_model = YOLO("yolov8n.pt", map_location="cpu")
+model_path = "yolov8n.pt"
+
+try:
+    torch.load(model_path, map_location="cpu")  
+    yolo_model = YOLO(model_path)
+except Exception as e:
+    print(f"模型載入錯誤: {e}")
 
 # 載入 Faster R-CNN 模型
 faster_rcnn_model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="COCO_V1")