Update app.py

app.py

@@ -1,12 +1,12 @@
-import gradio as gr
-import cv2
+# import gradio as gr
+import onnxruntime as ort
+import numpy as np
 from PIL import Image
 from torchvision import transforms
 import io
 import rdkit
-import numpy as np
+import cv2
 import torch
-import onnxruntime as ort
 from rdkit import Chem
 from rdkit.Chem import Draw
 from postprocessor import RTDETRPostProcessor
@@ -55,7 +55,7 @@ def predict(input_image):
     """
     try:
         # 加载和初始化ONNX模型
-        session = ort.InferenceSession("model.onnx")  # 替换为实际模型路径
+        session = ort.InferenceSession("/cadd_data/samba_share/from_docker/rt-detr/onnx/i2mDETR_model.onnx", providers=['CUDAExecutionProvider'])  # 替换为实际模型路径
         
         # 预处理图片
          # Example usage: #change thie image