Update app.py

app.py

@@ -65,14 +65,17 @@ def predict(image):
                     return json.dumps({"error": f"Error decoding base64 image: {e}"})
 
             # Try to fetch the image from a URL
-            elif isinstance(image, str) and image.startswith("http"):
-                try:
-                    response = requests.get(image)
-                    image = Image.open(BytesIO(response.content))
-                    print(f"Fetched image from URL: {image}")
-                except Exception as e:
-                    print(f"Error fetching image from URL: {e}")
-                    return json.dumps({"error": f"Error fetching image from URL: {e}"})
+            elif isinstance(image, str):
+                if image.startswith("http://") or image.startswith("https://"):
+                    try:
+                        response = requests.get(image)
+                        image = Image.open(BytesIO(response.content))
+                        print(f"Fetched image from URL: {image}")
+                    except Exception as e:
+                        print(f"Error fetching image from URL: {e}")
+                        return json.dumps({"error": f"Error fetching image from URL: {e}"})
+                else:
+                    return json.dumps({"error": "Invalid URL format. Please provide a valid URL starting with 'http://' or 'https://'."})
 
             # Try to load the image from a local file path
             elif isinstance(image, str) and os.path.isfile(image):