Update app.py

app.py

@@ -1,7 +1,16 @@
 import gradio as gr
+import matplotlib.pyplot as plt
+from PIL import Image
 from ultralytics import YOLO
 
-mod = YOLO('best(1).pt')
+model = YOLO('best(1).pt')
 
 def response(image):
-    
+    results = model(image)
+    # Plot results image
+    im_bgr = results.plot()  # BGR-order numpy array
+    im_rgb = im_bgr[..., ::-1]  # Convert BGR to RGB
+    return im_rgb
+
+iface = gr.Interface(fn=response, input="image", output="image")
+iface.launch()