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)
|