Update app.py

app.py

@@ -131,10 +131,10 @@ def inference(image_path, model_name, bbox_threshold):
     tensor_img = tensor_img.unsqueeze(0)
 
     
-
+    print(model_name)
     if model_name == "RF-DETR-B":
         model_path = "rfdetr.onnx"
-    else:
+    if model_name == "RF-DETR-L":
         model_path = "rfdetrl.onnx"
 
     
@@ -184,7 +184,7 @@ demo = gr.Interface(
     fn=inference,
     inputs=[
         gr.Image(type="filepath", label="Input Image"),
-        gr.Dropdown(["RF-DETR-L"], value="RF-DETR-L", label="Model"),
+        gr.Dropdown(["RF-DETR-L", "RF-DETR-B"], value="RF-DETR-B", label="Model"),
         gr.Slider(value=0.5, minimum=0.0, maximum=0.9, step=0.05, label="BBox threshold"),
     ],
     outputs=gr.Gallery(label="Output", preview=True, height=500),