jimipatel commited on
Commit
a2fa8eb
·
verified ·
1 Parent(s): 307c5b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ pipeline = FluxPipeline.from_pretrained(
19
  use_auth_token=True # This uses the environment token if not passed explicitly
20
  ).to("cuda")
21
 
22
- @spaces.GPU(duration=70)
23
  def generate(prompt, negative_prompt, width, height, sample_steps, lora_id):
24
  try:
25
  pipeline.load_lora_weights(lora_id)
 
19
  use_auth_token=True # This uses the environment token if not passed explicitly
20
  ).to("cuda")
21
 
22
+ @spaces.GPU(duration=30)
23
  def generate(prompt, negative_prompt, width, height, sample_steps, lora_id):
24
  try:
25
  pipeline.load_lora_weights(lora_id)