Update app.py

app.py

@@ -59,11 +59,22 @@ def predict_from_image(image):
     else:
         return {"error": "Unexpected class prediction."}
 
+# Function to predict from URL
+def predict_from_url(url):
+    try:
+        response = requests.get(url)
+        response.raise_for_status()  # Ensure the request was successful
+        image = Image.open(BytesIO(response.content))
+        return predict_from_image(image)
+    except Exception as e:
+        return {"error": f"Failed to process the URL: {str(e)}"}
+
 # Gradio interface
 iface = gr.Interface(
-    fn=lambda image: predict_from_image(image),
+    fn=lambda image, url: predict_from_image(image) if image else predict_from_url(url),
     inputs=[
         gr.Image(type="pil", label="Upload an Image"),
+        gr.Textbox(label="Or Enter an Image URL", placeholder="Provide a valid image URL"),
     ],
     outputs=gr.JSON(label="Prediction Result"),
     live=True,