LailaMB commited on
Commit
50ae3aa
·
1 Parent(s): 9f16d4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -3,7 +3,6 @@ import gradio as gr
3
  from huggingface_hub import hf_hub_download
4
  from PIL import Image
5
 
6
-
7
  yolov7_weights = hf_hub_download(repo_id="LailaMB/visual_pollution_detection", filename="best_640_rpoch56.pt")
8
 
9
  model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True) # local repo
 
3
  from huggingface_hub import hf_hub_download
4
  from PIL import Image
5
 
 
6
  yolov7_weights = hf_hub_download(repo_id="LailaMB/visual_pollution_detection", filename="best_640_rpoch56.pt")
7
 
8
  model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True) # local repo