update templates

Signed-off-by: AnyISalIn <anyisalin@gmail.com>

app.py

@@ -71,6 +71,17 @@ def create_ui():
                     training_refresh_button = gr.Button(value="Refresh Training Status")
                     training_refresh_json = gr.JSON()
 
+                def on_upload_training_images(files):
+                    if files is None:
+                        return 2000
+                    return min(2000, len(files) * 200)
+
+                training_images.change(
+                    inputs=[training_images],
+                    outputs=max_train_steps,
+                    fn=on_upload_training_images,
+                )
+
                 def train(novita_key, gender, base_model, training_name, max_train_steps, training_images):
                     training_images = [_.name for _ in training_images]
                     try:
@@ -233,6 +244,16 @@ def create_ui():
                     "style_height": 768,
                     "style_width": 512,
                     "style_reference_image": "./00003.jpg",
+                },
+                {
+                    "style_prompt": "Iron Man, close up, digital art, character concept, magical realism, warm golden light, sunset, Marvel Cinematic Universe, Hogwarts, high-resolution, vibrant colors, realistic rendering, key art, fantasy fashion, Iron Man suit integration, elegant and powerful pose",
+                    "style_negative_prompt": "BadDream_53202, UnrealisticDream_53204",
+                    "style_gender": "man",
+                    "style_model": "dreamshaper_8_93211.safetensors",
+                    "style_method": "controlnet-depth",
+                    "style_height": 768,
+                    "style_width": 512,
+                    "style_reference_image": "./00004.jpg",
                 }
             ]
 
@@ -299,6 +320,7 @@ def create_ui():
                             height=height,
                             model_name=style_model,
                             steps=30,
+                            sampler_name=Samplers.DPMPP_M_KARRAS,
                         )
                     elif style_method == "controlnet-depth":
                         req = Txt2ImgRequest(
@@ -308,6 +330,7 @@ def create_ui():
                             height=height,
                             model_name=style_model,
                             steps=30,
+                            sampler_name=Samplers.DPMPP_M_KARRAS,
                             controlnet_units=[
                                 ControlnetUnit(
                                     input_image=image_to_base64(style_reference_image),
@@ -327,6 +350,7 @@ def create_ui():
                             height=height,
                             model_name=style_model,
                             steps=30,
+                            sampler_name=Samplers.DPMPP_M_KARRAS,
                             controlnet_units=[
                                 ControlnetUnit(
                                     input_image=image_to_base64(style_reference_image),
@@ -346,6 +370,7 @@ def create_ui():
                             height=height,
                             model_name=style_model,
                             steps=30,
+                            sampler_name=Samplers.DPMPP_M_KARRAS,
                             controlnet_units=[
                                 ControlnetUnit(
                                     input_image=image_to_base64(style_reference_image),