Update app.py

app.py

@@ -66,8 +66,10 @@ def inference(image_path, model_name, bbox_threshold):
     ort_inputs = {
         'input': tensor_img.cpu().numpy()
     }
-    
-    model_path = "rfdetr.onnx"
+    if model_name == "RF-DETR-B":
+        model_path = "rfdetr.onnx"
+    else:
+        model_path = "rfdetrl.onnx"
 
     
     sess_options = onnxruntime.SessionOptions()