Update app.py

app.py

@@ -11,9 +11,11 @@ from data_manager import get_dog_description
 from urllib.parse import quote
 os.system('pip install ultralytics')
 from ultralytics import YOLO
+from PIL import ImageDraw
 
-# 下載YOLOv5預訓練模型
-model_yolo = YOLO('yolov8n.pt')  # 使用 YOLOv5 預訓練模型
+
+# 下載YOLOv8預訓練模型
+model_yolo = YOLO('yolov8n.pt')  # 使用 YOLOv8 預訓練模型
 
 
 dog_breeds = ["Afghan_Hound", "African_Hunting_Dog", "Airedale", "American_Staffordshire_Terrier",