Update app.py

app.py

@@ -26,6 +26,8 @@ def predict(
         weights = "checkpoints_model_final_imagenet_40k_synthetic.pth"
     elif model == "100k synthetic":
         weights = "checkpoints_model_final_imagenet_100k_synthetic.pth"
+    else:
+        weights = "checkpoints_model_final_imagenet_5k_synthetic.pth"
     cfg.MODEL.WEIGHTS=weights
     predictor = DefaultPredictor(cfg)
     im = cv2.imread(img.name)
@@ -44,7 +46,7 @@ examples = [
 ]
 
 
-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(["5k synthetic", "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)