Update app.py

app.py

@@ -8,20 +8,17 @@ from pathlib import Path
 
 detection_model = AutoDetectionModel.from_pretrained(
     model_type='ultralytics',
-    model_path="/workspace/runs/detect/train2/weights/last.pt",  # Replace with your model path
+    model_path="./DDR.pt",  # Replace with your model path
     confidence_threshold=0.01,
     device="cpu"  # Change to 'cuda:0' if you have a GPU
 )
 
-OUTPUT_PATH = "/workspace/pred_image.jpg"
-TEMP_PNG_PATH = "/workspace/pred_image.png"
+OUTPUT_PATH = "./outputs/pred_image.jpg"
+TEMP_PNG_PATH = "./outputs/pred_image.png"
 
 def run_inference(image):
-    input_path = "/workspace/input_image.jpg"
-    cv2.imwrite(input_path, cv2.cvtColor(image, cv2.COLOR_RGB2BGR))
-    
     result = get_sliced_prediction(
-        input_path,
+        image,
         detection_model,
         slice_height=256,
         slice_width=256,