jiuface commited on
Commit
3b7f155
1 Parent(s): 4c93f86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -57,7 +57,7 @@ class calculateDuration:
57
 
58
  @spaces.GPU(duration=120)
59
  @torch.inference_mode()
60
- def generate_image(prompt, lora_strings_json, adapter_names, steps, seed, cfg_scale, width, height, progress):
61
 
62
  lora_configs = None
63
  adapter_names = []
@@ -178,7 +178,7 @@ def run_lora(prompt, lora_strings_json, cfg_scale, steps, randomize_seed, seed,
178
  error_message = ""
179
  try:
180
  print("Start applying for zeroGPU resources")
181
- final_image = generate_image(prompt, lora_strings_json, adapter_names, steps, seed, cfg_scale, width, height, progress)
182
  except Exception as e:
183
  error_message = str(e)
184
  gr.Error(error_message)
 
57
 
58
  @spaces.GPU(duration=120)
59
  @torch.inference_mode()
60
+ def generate_image(prompt, lora_strings_json, steps, seed, cfg_scale, width, height, progress):
61
 
62
  lora_configs = None
63
  adapter_names = []
 
178
  error_message = ""
179
  try:
180
  print("Start applying for zeroGPU resources")
181
+ final_image = generate_image(prompt, lora_strings_json, steps, seed, cfg_scale, width, height, progress)
182
  except Exception as e:
183
  error_message = str(e)
184
  gr.Error(error_message)