Update app.py

app.py

@@ -56,7 +56,10 @@ def gradio_predict(url, threshold):
     tags, error = predict_tags(url, threshold)
     if error:
         return error, None
-    return "\n".join([f"{tag}: {score:.2f}" for tag, score in tags]), url
+    # Return tags, and the image as a PIL object
+    image = load_image_from_url(url)
+    return "\n".join([f"{tag}: {score:.2f}" for tag, score in tags]), image
+
 
 demo = gr.Interface(
     fn=gradio_predict,