Update app.py

app.py

@@ -202,10 +202,6 @@ def center_crop_image_as_square(img):
     
     img_cropped = img.crop((left, top, right, bottom))
     return img_cropped
-    
-def check_selected(selected_state):
-    if not selected_state:
-        raise gr.Error("You must select a style")
 
 def merge_incompatible_lora(full_path_lora, lora_scale):
     for weights_file in [full_path_lora]:
@@ -309,11 +305,14 @@ def generate_image(prompt, negative, face_emb, face_image, face_kps, image_stren
         
     return image
     
-def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, lora_repo_name):
+def run_lora(face_image, prompt, negative, lora_weight, face_strength, image_strength, guidance_scale, depth_control_scale, lora_repo_name):
     # get the lora and its default values
     lora = getLoraByRepoName(lora_repo_name)
     default_values = getLoraDefaultsByRepoName(lora_repo_name)
     
+    if not lora_repo_name:
+        raise gr.Error("You must input a LoRA repo name")
+
     st = time.time()
 
     face_image = readb64(face_image)
@@ -350,9 +349,6 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
     if negative == "":
         negative = None
 
-    if not selected_state:
-        raise gr.Error("You must select a LoRA")
-
     weight_name = lora["weights"]
     
     full_path_lora = state_dicts[lora["repo"]]["saved_name"]
@@ -364,7 +360,7 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
 
     st = time.time()
 
-    image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, lora, full_path_lora, lora_scale, st)
+    image = generate_image(prompt, negative, face_emb, face_image, face_kps, image_strength, guidance_scale, face_strength, depth_control_scale, lora, full_path_lora, lora_weight, st)
         
     image_base64 = writeb64(image)
         
@@ -389,7 +385,7 @@ with gr.Blocks() as demo:
     negative = gr.Textbox(label="Negative Prompt")
     
     # initial value was 0.9
-    weight = gr.Slider(0, 10, value=6, step=0.1, label="LoRA weight")
+    lora_weight = gr.Slider(0, 10, value=6, step=0.1, label="LoRA weight")
     
     # initial value was 0.85
     face_strength = gr.Slider(0, 1, value=0.75, step=0.01, label="Face strength", info="Higher values increase the face likeness but reduce the creative liberty of the models")
@@ -411,7 +407,7 @@ with gr.Blocks() as demo:
             input_image_base64,
             prompt,
             negative,
-            weight,
+            lora_weight,
             face_strength,
             image_strength,
             guidance_scale,