Update app.py

app.py

@@ -62,9 +62,8 @@ def inference(image_path, model_name, bbox_threshold):
     tensor_img, _ = transforms(image, None)
     tensor_img = tensor_img.unsqueeze(0)
 
-    ort_inputs = {
-        'input': tensor_img.cpu().numpy()
-    }
+    ort_inputs = {ort_session.get_inputs()[0].name: tensor_img.cpu().numpy()}
+
     if model_name == "RF-DETR-B":
         model_path = "rfdetr.onnx"
     else: