jibsn commited on
Commit
3633b1b
·
verified ·
1 Parent(s): 83dd5ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ def predict(input_image):
55
  """
56
  try:
57
  # 加载和初始化ONNX模型
58
- session = ort.InferenceSession("/cadd_data/samba_share/from_docker/rt-detr/onnx/i2mDETR_model.onnx", providers=['CUDAExecutionProvider']) # 替换为实际模型路径
59
 
60
  # 预处理图片
61
  # Example usage: #change thie image
 
55
  """
56
  try:
57
  # 加载和初始化ONNX模型
58
+ session = ort.InferenceSession("model.onnx")
59
 
60
  # 预处理图片
61
  # Example usage: #change thie image