Update app.py

app.py

@@ -35,7 +35,7 @@ async def generate_image_async(prompt, aspect_ratio):
                 response.raise_for_status()
                 response_data = await response.json()
                 if "output" in response_data:
-                    return response_data["outputFull"]
+                    return response_data
                 elif "error" in response_data:
                     return response_data['error']
                 else: