Spaces:

LailaMB
/

visual_pollution_detection

Sleeping

LailaMB commited on Jan 20, 2023

Commit

25f669d

1 Parent(s): c88766e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,9 +4,10 @@ from huggingface_hub import hf_hub_download
 from PIL import Image
 REPO_ID = "Yolov7/yolov7"
-FILENAME = "models/best_640_rpoch56.pt"
-yolov7_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
 model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True)  # local repo

 from PIL import Image
 REPO_ID = "Yolov7/yolov7"
+FILENAME =
+yolov7_weights = hf_hub_download(repo_id="LailaMB/visual_pollution_detection", filename="best_640_rpoch56.pt")
 model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True)  # local repo