Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,14 +4,14 @@ from huggingface_hub import hf_hub_download
|
|
4 |
from PIL import Image
|
5 |
|
6 |
REPO_ID = "LailaMB/pollution_detector"
|
7 |
-
FILENAME = "best_640_rpoch56.pt"
|
8 |
|
9 |
yolov7_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
|
10 |
|
11 |
model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True) # local repo
|
12 |
|
13 |
def object_detection(im, size=640):
|
14 |
-
|
15 |
#results.print() # print results to screen
|
16 |
#results.show() # display results
|
17 |
#results.save() # save as results1.jpg, results2.jpg... etc.
|
|
|
4 |
from PIL import Image
|
5 |
|
6 |
REPO_ID = "LailaMB/pollution_detector"
|
7 |
+
FILENAME = "models/best_640_rpoch56.pt"
|
8 |
|
9 |
yolov7_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
|
10 |
|
11 |
model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', path=yolov7_weights, force_reload=True) # local repo
|
12 |
|
13 |
def object_detection(im, size=640):
|
14 |
+
results = model(im) # inference
|
15 |
#results.print() # print results to screen
|
16 |
#results.show() # display results
|
17 |
#results.save() # save as results1.jpg, results2.jpg... etc.
|