Update app.py

app.py

@@ -9,22 +9,13 @@ openai.api_key = os.getenv("OPENAI_API_KEY")
 
 
 # Define the function that generates the image
-def generate_image(prompt):
-    response = requests.post("https://api.openai.com/v1/images/generations", json={
-        "model": "image-alpha-001",
-        "prompt": prompt,
-        "num_images": 4,
-        "size": "1024x1024",
-        "response_format": "url"
-    }, headers={
-        "Content-Type": "application/json",
-        "Authorization": "Bearer "
-    })
-    response.raise_for_status()
-    image_url = response.json()["data"][0]["url"]
-    image = Image.open(BytesIO(requests.get(image_url).content))
-    return image
-
+response = openai.Image.create(
+  prompt=prompt,
+  n=1,
+  size="1024x1024"
+)
+image_url = response['data'][0]['url']
+ 
 
 iface = gr.Interface(
     fn=generate_image,