owaiskha9654 commited on
Commit
89bb59b
1 Parent(s): 25387ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ from PIL import Image
6
  REPO_ID = "owaiskha9654/Yolov7_Custom_Object_Detection"
7
  FILENAME = "best.pt"
8
 
 
9
  yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
10
 
11
  model = torch.hub.load('Owaiskhan9654/yolov7-1:main',model='custom', path_or_model=yolov7_custom_weights, force_reload=True) # My Github repository https://github.com/Owaiskhan9654
 
6
  REPO_ID = "owaiskha9654/Yolov7_Custom_Object_Detection"
7
  FILENAME = "best.pt"
8
 
9
+
10
  yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
11
 
12
  model = torch.hub.load('Owaiskhan9654/yolov7-1:main',model='custom', path_or_model=yolov7_custom_weights, force_reload=True) # My Github repository https://github.com/Owaiskhan9654