Update app.py

app.py

@@ -22,9 +22,9 @@ def predict(
     model,
     img
 ):
-    if model=="40k_synthetic":
+    if model=="40k synthetic":
         weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
-    else:
+    elif model == "100k synthetic":
         weights = "checkpoints_model_final_imagenet_100k_synthetic.pth"
     cfg.MODEL.WEIGHTS=weights
     predictor = DefaultPredictor(cfg)
@@ -36,12 +36,12 @@ def predict(
 title = "Indoor Pet Detection"
 description = "Demo for Indoor Pet Detection"
 examples = [
-    ["100k_synthetic", 'example.jpg'],
-    ["40k_synthetic", 'example.jpg']
+    ["100k synthetic", 'example.jpg'],
+    ["40k synthetic", 'example.jpg']
 ]
 
 
-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,
+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(debug=True)