Vvaann commited on
Commit
8382c5c
·
verified ·
1 Parent(s): af84e37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -7,10 +7,14 @@ from PIL import Image
7
  from pytorch_grad_cam import GradCAM
8
  from pytorch_grad_cam.utils.image import show_cam_on_image
9
  import gradio as gr
10
- from huggingface_hub import hf_hub_download
11
  from models.yolo import Model
12
 
13
- model = Model('best.pt')
 
 
 
 
 
14
 
15
  classes = ('ball', 'goalkeeper', 'player', 'referee')
16
 
 
7
  from pytorch_grad_cam import GradCAM
8
  from pytorch_grad_cam.utils.image import show_cam_on_image
9
  import gradio as gr
 
10
  from models.yolo import Model
11
 
12
+ from huggingface_hub import hf_hub_download
13
+ hf_hub_download("merve/yolov9', local_dir="./")
14
+
15
+ import yolov9
16
+
17
+ model = yolov9.load('best.pt')
18
 
19
  classes = ('ball', 'goalkeeper', 'player', 'referee')
20