Spaces:

Bonosa2
/

dall-e_image-generation

Runtime error

bonosa commited on May 23, 2023

Commit

3073b66

1 Parent(s): f66daf5

4

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ def generate_image(prompt):
     response = openai.Image.create(
         prompt=prompt,
         n=1,
-        size="1024x1024"
     )
     image_url = response['data'][0]['url']
@@ -19,7 +19,7 @@ def generate_image(prompt):
         with open('temp.jpg', 'wb') as f:
             f.write(url.read())
         img = Image.open('temp.jpg')
-        img = img.resize((512, 512))
     return img

     response = openai.Image.create(
         prompt=prompt,
         n=1,
+        size="256x256"
     )
     image_url = response['data'][0]['url']
         with open('temp.jpg', 'wb') as f:
             f.write(url.read())
         img = Image.open('temp.jpg')
+        img = img.resize((256, 256))
     return img