Update app.py

app.py

@@ -26,7 +26,7 @@ def process_img(image, filename):
 
     # filename = image.name.split("/")[-1]
     # if filename in image_to_weights:
-    model_path = image_to_weights[image_to_weights[1]]
+    model_path = image_to_weights[1]
     checkpoints = torch.load(model_path, map_location=device)
     model.load_state_dict(checkpoints['params'])
     model.to(device)