hippoiam10 commited on
Commit
db6f67c
·
verified ·
1 Parent(s): bb481bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -7,7 +7,12 @@ from PIL import Image, ImageDraw
7
  import numpy as np
8
  from sklearn.metrics import precision_recall_fscore_support
9
 
10
- yolo_model = YOLO("yolov8n")
 
 
 
 
 
11
 
12
 
13
 
 
7
  import numpy as np
8
  from sklearn.metrics import precision_recall_fscore_support
9
 
10
+ model_url = "https://huggingface.co/Ultralytics/YOLOv8/resolve/main/yolov8n.pt"
11
+ model_path = "yolov8n.pt"
12
+
13
+ # 下載模型
14
+ urllib.request.urlretrieve(model_url, model_path)
15
+ yolo_model = YOLO(model_path)
16
 
17
 
18