Upload app.py

app.py

@@ -86,6 +86,14 @@ def preprocess_image(image):
     """
     Preprocess image for model prediction.
     """
+    # Handle different input types
+    if isinstance(image, str):
+        # If it's a file path or URL, load it
+        image = Image.open(image)
+    elif not isinstance(image, Image.Image):
+        # If it's a numpy array, convert to PIL
+        image = Image.fromarray(image)
+
     # Convert to RGB if necessary
     if image.mode != 'RGB':
         image = image.convert('RGB')