Update app.py

app.py

@@ -1,7 +1,8 @@
-
 import gradio as gr
 import torch
 from ultralyticsplus import YOLO, render_result
+from PIL import Image
+import os
 
 def yolov8_func(image, 
                 image_size, 
@@ -21,9 +22,13 @@ def yolov8_func(image,
     print("Confidence: ", box.conf)
     print("Coordinates: ", box.xyxy)
 
-    # Render and return the result
+    # Render the result
     render = render_result(model=model, image=image, result=result[0])
-    return render
+    
+    # Save the rendered image locally
+    save_path = "predicted_image.jpg"  # Specify the output path
+    render.save(save_path)  # Save using PIL's save method
+    return save_path
 
 # Define inputs for the Gradio app
 inputs = [