Nick088 commited on
Commit
fb166bb
·
verified ·
1 Parent(s): c2a1a9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=90)
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)