Update app.py

app.py

@@ -43,7 +43,10 @@ def object_detection(image):
     buf.seek(0)
     plt.close()
 
-    return buf
+    # Convert buffer to an Image object
+    result_image = Image.open(buf)
+
+    return result_image
 
 # Define the Gradio interface
 demo = gr.Interface(