Update app.py

app.py

@@ -48,7 +48,11 @@ transform = transforms.Compose([
 
 def predict(data):
     try:
-        image_input = data.get('image', None)
+        # Expecting data to be a list
+        if not isinstance(data, list) or len(data) == 0:
+            return json.dumps({"error": "Input data should be a non-empty list."})
+        
+        image_input = data[0].get('image', None)
         if not image_input:
             return json.dumps({"error": "No image provided."})
 
@@ -109,3 +113,4 @@ iface = gr.Interface(
 
 # Launch the Gradio interface 
 iface.launch(share=True, show_error=True)
+