Update app.py

app.py

@@ -26,7 +26,12 @@ def predict(img):
         if isinstance(img, Image.Image):
             img = np.array(img)
 
-        # If the input is still not a NumPy array, return an error
+        # If the image is base64, convert it to a PIL image and then to a NumPy array
+        elif isinstance(img, str):  # base64 image
+            img = Image.open(io.BytesIO(base64.b64decode(img.split(',')[1])))
+            img = np.array(img)
+
+        # If the input is still not a NumPy array, raise an error
         if not isinstance(img, np.ndarray):
             raise ValueError("Input is not a valid image")