Update app.py

app.py

@@ -207,16 +207,16 @@ def merge_incompatible_lora(full_path_lora, lora_scale):
 
 def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_strength, image_strength, guidance_scale, depth_control_scale, sdxl_loras, progress=gr.Progress(track_tqdm=True)):
     global last_lora, last_merged, last_fused, pipe
-    #print("Images:", images)
-    #print("Face image", images[0])
-    #face_image = images[0]
+
     face_image = center_crop_image_as_square(face_image)
-    #print("Cropped image:", face_image)
-    face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
-    face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
-    face_emb = face_info['embedding']
-    face_kps = draw_kps(face_image, face_info['kps'])
-    
+    try:
+        face_info = app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
+        face_info = sorted(face_info, key=lambda x:(x['bbox'][2]-x['bbox'][0])*x['bbox'][3]-x['bbox'][1])[-1] # only use the maximum face
+        face_emb = face_info['embedding']
+        face_kps = draw_kps(face_image, face_info['kps'])
+    except: 
+        raise gr.Error("No face found in your image. Only face images work here. Try again")
+        
     for lora_list in lora_defaults:
         if lora_list["model"] == sdxl_loras[selected_state.index]["repo"]:
             prompt_full = lora_list.get("prompt", None)
@@ -224,7 +224,9 @@ def run_lora(face_image, prompt, negative, lora_scale, selected_state, face_stre
                 prompt = prompt_full.replace("<subject>", prompt)
 
     
-    print("Prompt:", prompt)    
+    print("Prompt:", prompt) 
+    if(prompt == ""):
+        prompt = "A person"
     #prepare face zoe
     with torch.no_grad():
         image_zoe = zoe(face_image)
@@ -347,7 +349,7 @@ with gr.Blocks(css="custom.css") as demo:
                 )
         with gr.Column(scale=3):
             with gr.Row():
-                prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, placeholder="A person", elem_id="prompt")
+                prompt = gr.Textbox(label="Prompt", show_label=False, lines=1, max_lines=1, info="Describe your subject (optional)", value="A person", elem_id="prompt")
                 button = gr.Button("Run", elem_id="run_button")
             with gr.Group(elem_id="share-btn-container", visible=False) as share_group:
                 community_icon = gr.HTML(community_icon_html)