Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
|
|
| 30 |
model.to(device)
|
| 31 |
|
| 32 |
# Define the image generation function
|
| 33 |
-
@spaces.GPU(duration=
|
| 34 |
def generate_image(prompt, enhance_prompt, negative_prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt):
|
| 35 |
if seed == 0:
|
| 36 |
seed = random.randint(1, 2**32-1)
|
|
|
|
| 30 |
model.to(device)
|
| 31 |
|
| 32 |
# Define the image generation function
|
| 33 |
+
@spaces.GPU(duration=60)
|
| 34 |
def generate_image(prompt, enhance_prompt, negative_prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt):
|
| 35 |
if seed == 0:
|
| 36 |
seed = random.randint(1, 2**32-1)
|