Update app.py

app.py

@@ -4,14 +4,14 @@ from huggingface_hub import hf_hub_download
 from PIL import Image
 
 REPO_ID = "LailaMB/pollution_detector"
-FILENAME = "best_640_rpoch56.pt"
+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
 
 def object_detection(im, size=640):
- 	results = model(im)  # inference
+    results = model(im)  # inference
     #results.print() # print results to screen
     #results.show()  # display results
     #results.save()  # save as results1.jpg, results2.jpg... etc.