Update app.py

app.py

@@ -1,18 +1,14 @@
 import gradio as gr
-import torch
+from ultralytics import YOLO
 import cv2
 import numpy as np
-from ultralytics import YOLO
-
-def load_model(model_path="ddr.pt"):
-    return YOLO(model_path)
-
-model = load_model()
 
-def predict(image):
-    results = model(image, conf=0.1)
-    pred_img = results[0].plot()  # Visualize detections
-    return pred_img
+def detect_objects(image):
+    model = YOLO("./ddr.pt")  # Load the trained model
+    results = model(image, conf=0.10)  # Run inference with confidence threshold
+    
+    output_image = results[0].plot()  # Get the image with bounding boxes
+    return output_image
 
 iface = gr.Interface(
     fn=predict,