Update app.py

app.py

@@ -1,6 +1,6 @@
 import os
 import gradio as gr
-from io import BytesIO
+#from io import BytesIO
 from PIL import Image
 from cloudflare import Cloudflare
 #from typing import List, Union
@@ -91,15 +91,13 @@ client = Cloudflare(api_token=API_TOKEN)
 
 def generate_image(prompt):
     try:
-        image = client.workers.ai.with_raw_response.run(
+        data = client.workers.ai.with_raw_response.run(
             model_name=IMAGE_MODEL,
             account_id=ACCOUNT_ID,
             prompt=prompt,
         )
 
-        image = Image.open(BytesIO(image.read()))
-
-        image = image.resize((512, 512))
+        image = Image.open(data)
 
         print(f'image: {image}')