Update app.py

app.py

@@ -147,8 +147,15 @@ def process_image(image_input):
     # Initialize the Gradio client with the URL of the Gradio server
     client = Client("https://adept-fuyu-8b-demo.hf.space/--replicas/pqjvl/")
 
-    # Check if the image input is a base64 string, file path (str), or a PIL Image
-    if isinstance(image_input, str):
+    # Check if the image input is a NumPy array
+    if isinstance(image_input, np.ndarray):
+        # Convert the NumPy array to a PIL Image
+        image = Image.fromarray(image_input)
+        # Save the PIL Image to a temporary file
+        with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp_file:
+            image.save(tmp_file.name)
+            image_path = tmp_file.name
+    elif isinstance(image_input, str):
         try:
             # Try to decode if it's a base64 string
             image = decode_image(image_input)