Update app.py

app.py

@@ -53,8 +53,9 @@ def generate(region1_concept,
     region1 = pose_image['region1']
     region2 = pose_image['region2']
 
-    region1_prompt = f'[<{region1_concept}1> <{region1_concept}2>, {region1_prompt}]'
-    region2_prompt = f'[<{region2_concept}1> <{region2_concept}2>, {region2_prompt}]'
+    region_pos_prompt = "high resolution, best quality, highly detailed, sharp focus, expressive, 8k uhd, detailed, sophisticated"
+    region1_prompt = f'<{region1_concept}1> <{region1_concept}2>, {region1_prompt}, {region_pos_prompt}'
+    region2_prompt = f'<{region2_concept}1> <{region2_concept}2>, {region2_prompt}, {region_pos_prompt}'
     prompt_rewrite=f"{region1_prompt}-*-{region_neg_prompt}-*-{region1}|{region2_prompt}-*-{region_neg_prompt}-*-{region2}"
     print(prompt_rewrite)
 
@@ -117,6 +118,7 @@ def merge(concept1, concept2):
     modelbase_path = os.path.join(save_path,'combined_model_base')
     assert os.path.isdir(modelbase_path)
 
+    # save_path = 'experiments/multi-concept/elsa_moana_weight18/combined_model_base'
     return modelbase_path
 
 def infer(pretrained_model,
@@ -180,19 +182,18 @@ def infer(pretrained_model,
     
     return image[0]
 
-
 def on_select(evt: gr.SelectData):  # SelectData is a subclass of EventData
     return evt.value['image']['orig_name']
 
 examples_context = [
     'walking at Stanford university campus',
-    'in a castle',
+    'in front of a castle',
     'in the forest',
-    'in front of Eiffel tower'
+    'in the style of cyberpunk'
 ]
 
-examples_region1 = ['wearing red hat, high resolution, best quality']
-examples_region2 = ['smilling, wearing blue shirt, high resolution, best quality']
+examples_region1 = ['wearing a red hat']
+examples_region2 = ['smiling, wearing a blue shirt']
 
 with open('multi-concept/pose_data/pose.json') as f:
     d = json.load(f)
@@ -231,15 +232,15 @@ with gr.Blocks(css=css) as demo:
                 
 
                 with gr.Row():
-
-                    character_list = ["Elsa", "Moana", "Woody", "Rapunzel"]
+                    
+                    concept_list = ["Elsa", "Moana", "Woody", "Rapunzel", "Elastigirl"]
                     region1_concept = gr.Dropdown(
-                        character_list,
+                        concept_list,
                         label="Character 1",
                         info="Will add more characters later!"
                     )
                     region2_concept = gr.Dropdown(
-                        character_list, 
+                        concept_list, 
                         label="Character 2", 
                         info="Will add more characters later!"
                     )
@@ -337,7 +338,7 @@ with gr.Blocks(css=css) as demo:
                     
                     keypose_adaptor_weight = gr.Slider(
                             label="Keypose Adapter Weight",
-                            minimum = 0,
+                            minimum = 0.1,
                             maximum = 1,
                             step= 0.01,
                             value=1.0,