Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
 import requests
+import numpy as np
+from PIL import Image  # Needed to convert binary data to numpy array
 
 # Function to call the API and generate the image
 def generate_image(content, style):
@@ -12,18 +14,15 @@ def generate_image(content, style):
     }
     response = requests.post('https://api.fliki.ai/v1/generate/text-to-image', headers=headers, json=json_data)
     
-    # Handle the response here based on the API's response format
-    # If the response is a URL:
-    # return response.json()['url']
-
-    # If the response is binary data:
-    return response.content
+    # Convert binary data to numpy array
+    image = Image.open(BytesIO(response.content))
+    return np.array(image)
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=generate_image,
     inputs=[gr.Textbox(label="Content"), gr.Textbox(label="Style")],
-    outputs=gr.Image(type="auto"),  # 'auto' will automatically handle URLs or binary data
+    outputs=gr.Image(type="numpy"),  # Use numpy type for image output
     title="Text to Image Generator",
     description="Enter a description and style to generate an image."
 )