Update app.py

app.py

@@ -40,7 +40,7 @@ def predict(img):
         detection_threshold=0.5
     )
     
-    return pred_dict["img"]
+    return pred_dict_2["img"]
     
 gr.Interface(
     fn=predict,