Update app.py

app.py

@@ -23,8 +23,8 @@ def generate_image(prompt, model_name):
     client = InferenceClient(model_name)
     response = client.text_to_image(prompt)
     #image_data = response[0]['image']
-    image = Image.open(BytesIO(base64.b64decode(response)))
-    return image
+    #image = Image.open(BytesIO(base64.b64decode(response)))
+    return response
     if isinstance(response, list):
         image_data = response[0]['image']
         image = Image.open(BytesIO(base64.b64decode(response)))