Update app.py

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 
 yolov7_weights = hf_hub_download(repo_id="LailaMB/visual_pollution_detection", filename="best_640_rpoch56.pt")
 
-model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', "LailaMB/visual_pollution_detection/best_640_rpoch56.pt", force_reload=True)  # local repo
+model = torch.hub.load('WongKinYiu/yolov7:main', 'custom', yolov7_weights, force_reload=True)  # local repo
 
 def object_detection(im, size=640):
     results = model(im)  # inference