yonishafir commited on
Commit
82fcc2f
·
verified ·
1 Parent(s): 2c91a3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -227,7 +227,8 @@ lora_names = Loras_dict.keys()
227
  def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_images, ip_adapter_scale, kps_scale, canny_scale, lora_name, lora_scale, progress=gr.Progress(track_tqdm=True)):
228
  # def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_images, ip_adapter_scale, kps_scale, canny_scale, progress=gr.Progress(track_tqdm=True)):
229
  global CURRENT_LORA_NAME # Use the global variable to track LoRA
230
-
 
231
  if image_path is None:
232
  raise gr.Error(f"Cannot find any input face image! Please upload a face image.")
233
 
 
227
  def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_images, ip_adapter_scale, kps_scale, canny_scale, lora_name, lora_scale, progress=gr.Progress(track_tqdm=True)):
228
  # def generate_image(image_path, prompt, num_steps, guidance_scale, seed, num_images, ip_adapter_scale, kps_scale, canny_scale, progress=gr.Progress(track_tqdm=True)):
229
  global CURRENT_LORA_NAME # Use the global variable to track LoRA
230
+ CURRENT_LORA_NAME = None
231
+
232
  if image_path is None:
233
  raise gr.Error(f"Cannot find any input face image! Please upload a face image.")
234