Update app.py

app.py

@@ -15,8 +15,11 @@ def generate_image(prompt):
       n=1,
       size="1024x1024"
     )    
+    response.raise_for_status()
+    image_url = response.json()["data"][0]["url"]
+    image = Image.open(BytesIO(requests.get(image_url).content))
+    return image
 
-image_url = response['data'][0]['url']
  
 
 iface = gr.Interface(