Update app.py

app.py

@@ -45,7 +45,7 @@ def predict_clothing(images):
 
 @spaces.GPU(duration=180)
 def generate_image(category, img1, img2, img3, height, width, img_guidance_scale, inference_steps, seed, separate_cfg_infer, offload_model,
-                   use_input_image_size_as_output, max_input_image_size, randomize_seed, guidance_scale=2.6):
+                   use_input_image_size_as_output, max_input_image_size, randomize_seed, guidance_scale=2.5):
 
     print()
     input_images = [img1, img2, img3]
@@ -57,10 +57,10 @@ def generate_image(category, img1, img2, img3, height, width, img_guidance_scale
     wears = predict_clothing(input_images[1:])
     if len(wears)==1:
         dress = wears[0]
-        text = f"""A {category} in <img><|image_1|></img> wearing {dress} in <img><|image_2|></img>."""
+        text = f"""The {category} in <img><|image_1|></img> wearing {dress} in <img><|image_2|></img>."""
     elif len(wears)==2:
         topwear, bottomwear = wears[0], wears[1]
-        text = f"""A {category} in <img><|image_1|></img> wearing {topwear} in <img><|image_2|></img> and {bottomwear} in <img><|image_3|></img>."""
+        text = f"""The {category} in <img><|image_1|></img> wearing {topwear} in <img><|image_2|></img> and {bottomwear} in <img><|image_3|></img>."""
     else:
         input_images = None
 
@@ -140,8 +140,8 @@ with gr.Blocks() as demo:
             category = gr.Radio(["man", "woman", "boy", "girl"], label="Category", info="Choose one category from the following")
             
             # sliders            
-            height_input = gr.Slider(label="Height", minimum=128, maximum=1024, value=768, step=16)
-            width_input = gr.Slider(label="Width", minimum=128, maximum=1024, value=432, step=16)
+            height_input = gr.Slider(label="Height", minimum=128, maximum=1024, value=512, step=16)
+            width_input = gr.Slider(label="Width", minimum=128, maximum=1024, value=512, step=16)
     
             # guidance_scale_input = gr.Slider(label="Guidance Scale", minimum=1.0, maximum=5.0, value=2.5, step=0.1)
     
@@ -151,7 +151,7 @@ with gr.Blocks() as demo:
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
 
         with gr.Column():
-            max_input_image_size = gr.Slider(label="max_input_image_size", minimum=128, maximum=2048, value=768, step=16)
+            max_input_image_size = gr.Slider(label="max_input_image_size", minimum=128, maximum=2048, value=1024, step=16)
             
             img_guidance_scale_input = gr.Slider(label="img_guidance_scale", minimum=1.0, maximum=2.0, value=1.6, step=0.1)