Update app.py

app.py

@@ -62,7 +62,7 @@ def inference(image_path, model_name, bbox_threshold):
     tensor_img, _ = transforms(image, None)
     tensor_img = tensor_img.unsqueeze(0)
 
-    ort_inputs = {ort_session.get_inputs()[0].name: tensor_img.cpu().numpy()}
+    
 
     if model_name == "RF-DETR-B":
         model_path = "rfdetr.onnx"
@@ -77,8 +77,8 @@ def inference(image_path, model_name, bbox_threshold):
         providers=["CUDAExecutionProvider", "CPUExecutionProvider"],
         sess_options=sess_options
     )
-
-    ort_inputs = {ort_session.get_inputs()[0].name: img_transformed}
+    
+    ort_inputs = {ort_session.get_inputs()[0].name: tensor_img.cpu().numpy()}
     ort_outs = ort_session.run(None, ort_inputs)
 
     boxes, labels, scores = ort_outs