Update app.py
Browse files
app.py
CHANGED
@@ -204,7 +204,7 @@ def resize_keep_aspect_ratio(pil_image, target_size=1024):
|
|
204 |
return pil_image.resize((new_W, new_H))
|
205 |
|
206 |
|
207 |
-
@spaces.GPU(
|
208 |
def generate_image(
|
209 |
prompt,
|
210 |
image_1, caption_1,
|
|
|
204 |
return pil_image.resize((new_W, new_H))
|
205 |
|
206 |
|
207 |
+
@spaces.GPU()
|
208 |
def generate_image(
|
209 |
prompt,
|
210 |
image_1, caption_1,
|