Update app.py

app.py

@@ -1,11 +1,6 @@
 import base64
 import gradio as gr
-import io
-import openai
-import os
 import random
-import requests
-from PIL import Image
 from fastai.vision.all import *
 from openai import OpenAI
 from pathlib import Path
@@ -116,13 +111,13 @@ def process_image(img):
 
         result = client.images.generate(
             model="gpt-image-1",
-            prompt="Draw a 2D pixel art style sprite sheet of a tabby gray cat",
+            prompt=random.choice(prompt_templates).format(flower=predicted_class),
             size="1024x1024",
             background="transparent",
-            quality="high",
+            quality="medium"
         )
 
-        image_base64 = result.json()["data"][0]["b64_json"]
+        image_base64 = result.data[0].b64_json
         generated_image = base64.b64decode(image_base64)
 
     except Exception as e: