Update app.py

app.py

@@ -36,16 +36,27 @@ transform = transforms.Compose([
 # Prediction function
 def process_image(image, image_url=None):
     try:
+        # Ensure that the image is not None
+        if image is None and not image_url:
+            return "No image or URL provided."
+
         # Handle URL-based image loading
         if image_url:
-            response = requests.get(image_url)
-            image = Image.open(BytesIO(response.content))
+            try:
+                response = requests.get(image_url)
+                response.raise_for_status()  # Raise an error if the request fails
+                image = Image.open(BytesIO(response.content))
+            except Exception as e:
+                return f"Error fetching image from URL: {e}"
 
         # Handle local file path image loading (Gradio File input)
         elif isinstance(image, str) and os.path.isfile(image):
-            image = Image.open(image)
+            try:
+                image = Image.open(image)
+            except Exception as e:
+                return f"Error loading image from local path: {e}"
 
-        # Validate image
+        # Validate that the image is loaded correctly
         if not isinstance(image, Image.Image):
             return "Invalid image format received."