Update app.py

app.py

@@ -19,13 +19,13 @@ cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
 # cfg.MODEL.WEIGHTS = "checkpoints_model_final_imagenet_40k_synthetic.pth"
 
 def predict(
-    img,
-    model="40k_synthetic"
+    model,
+    img
 ):
     if model=="40k_synthetic":
         weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
     else:
-        weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
+        weights = "checkpoints_model_final_imagenet_100k_synthetic.pth"
     cfg.MODEL.WEIGHTS=model
     predictor = DefaultPredictor(cfg)
     im = cv2.imread(img.name)
@@ -35,10 +35,10 @@ def predict(
 
 title = "Pet Detection"
 description = "Demo for Indoor Pet Detection"
-examples = [['example.jpg']]
+examples = [["100k_synthetic", 'example.jpg']]
 
 
-gr.Interface(predict, inputs=[gr.inputs.Image(type="file"), gr.inputs.Dropdown(["40k_synthetic", "100k_synthetic"])], outputs=gr.outputs.Image(type="pil"),enable_queue=True, title=title,
+gr.Interface(predict, [gr.inputs.Dropdown(["40k_synthetic", "100k_synthetic"]), gr.inputs.Image(type="file")], outputs=gr.outputs.Image(type="pil"),enable_queue=True, title=title,
     description=description,
     # article=article,
-    examples=examples).launch()
+    examples=examples).launch(debug=True)