owaiskha9654
commited on
Commit
·
be1ba3d
1
Parent(s):
2fe4d4a
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ FILENAME = "best.pt"
|
|
9 |
|
10 |
yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
|
11 |
|
12 |
-
model = torch.hub.load('',model='custom', path_or_model=yolov7_custom_weights, force_reload=True) #
|
13 |
|
14 |
def object_detection(im, size=640):
|
15 |
results = model(im)
|
|
|
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) # Github repository https://github.com/Owaiskhan9654
|
13 |
|
14 |
def object_detection(im, size=640):
|
15 |
results = model(im)
|