Update app.py

app.py

@@ -38,7 +38,7 @@ def predict_clothing(images):
 
 @spaces.GPU(duration=180)
 def generate_image(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=3.5):
+                   use_input_image_size_as_output, max_input_image_size, randomize_seed, guidance_scale=3):
     
     input_images = [img1, img2, img3]
     # Delete None
@@ -68,215 +68,34 @@ def generate_image(img1, img2, img3, height, width, img_guidance_scale, inferenc
     img = output[0]
     return img
 
+example_text = 
 def get_example():
     case = [
-        [
-            "A curly-haired man in a red shirt is drinking tea.",
-            None,
-            None,
-            None,
-            1024,
-            1024,
-            2.5,
-            1.6,
-            0,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "The woman in <img><|image_1|></img> waves her hand happily in the crowd",
-            "./imgs/test_cases/zhang.png",
-            None,
-            None,
-            1024,
-            1024,
-            2.5,
-            1.9,
-            128,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "A man in a black shirt is reading a book. The man is the right man in <img><|image_1|></img>.",
-            "./imgs/test_cases/two_man.jpg",
-            None,
-            None,
-            1024,
-            1024,
-            2.5,
-            1.6,
-            0,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "Two woman are raising fried chicken legs in a bar. A woman is <img><|image_1|></img>. Another woman is <img><|image_2|></img>.",
-            "./imgs/test_cases/mckenna.jpg",
-            "./imgs/test_cases/Amanda.jpg",
-            None,
-            1024,
-            1024,
-            2.5,
-            1.8,
-            65,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "A man and a short-haired woman with a wrinkled face are standing in front of a bookshelf in a library. The man is the man in the middle of <img><|image_1|></img>, and the woman is oldest woman in <img><|image_2|></img>",
-            "./imgs/test_cases/1.jpg",
-            "./imgs/test_cases/2.jpg",
-            None,
-            1024,
-            1024,
-            2.5,
-            1.6,
-            60,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "A man and a woman are sitting at a classroom desk. The man is the man with yellow hair in <img><|image_1|></img>. The woman is the woman on the left of <img><|image_2|></img>",
-            "./imgs/test_cases/3.jpg",
-            "./imgs/test_cases/4.jpg",
-            None,
-            1024,
-            1024,
-            2.5,
-            1.8,
-            66,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "The flower <img><|image_1|></img> is placed in the vase which is in the middle of <img><|image_2|></img> on a wooden table of a living room",
-            "./imgs/test_cases/rose.jpg",
-            "./imgs/test_cases/vase.jpg",
-            None,
-            1024,
-            1024,
-            2.5,
-            1.6,
-            0,
-            1024,
-            False,
-            False,
-        ],
-        [
-            "<img><|image_1|><img>\n Remove the woman's earrings. Replace the mug with a clear glass filled with sparkling iced cola.",
-            "./imgs/demo_cases/t2i_woman_with_book.png",
-            None,
-            None,
-            None,
-            None,
-            2.5,
-            1.6,
-            222,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "Detect the skeleton of human in this image: <img><|image_1|></img>.",
-            "./imgs/test_cases/control.jpg",
-            None,
-            None,
-            1024,
-            1024,
-            2.0,
-            1.6,
-            0,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "Generate a new photo using the following picture and text as conditions: <img><|image_1|><img>\n A young boy is sitting on a sofa in the library, holding a book. His hair is neatly combed, and a faint smile plays on his lips, with a few freckles scattered across his cheeks. The library is quiet, with rows of shelves filled with books stretching out behind him.",
-            "./imgs/demo_cases/skeletal.png",
-            None,
-            None,
-            1024,
-            1024,
-            2,
-            1.6,
-            999,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "Following the pose of this image <img><|image_1|><img>, generate a new photo: A young boy is sitting on a sofa in the library, holding a book. His hair is neatly combed, and a faint smile plays on his lips, with a few freckles scattered across his cheeks. The library is quiet, with rows of shelves filled with books stretching out behind him.",
-            "./imgs/demo_cases/edit.png",
-            None,
-            None,
-            1024,
-            1024,
-            2.0,
-            1.6,
-            123,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "Following the depth mapping of this image <img><|image_1|><img>, generate a new photo: A young girl is sitting on a sofa in the library, holding a book. His hair is neatly combed, and a faint smile plays on his lips, with a few freckles scattered across his cheeks. The library is quiet, with rows of shelves filled with books stretching out behind him.",
-            "./imgs/demo_cases/edit.png",
-            None,
-            None,
-            1024,
-            1024,
-            2.0,
-            1.6,
-            1,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "<img><|image_1|><\/img> What item can be used to see the current time? Please highlight it in blue.",
-            "./imgs/test_cases/watch.jpg",
-            None,
-            None,
-            1024,
-            1024,
-            2.5,
-            1.6,
-            666,
-            1024,
-            False,
-            True,
-        ],
-        [
-            "According to the following examples, generate an output for the input.\nInput: <img><|image_1|></img>\nOutput: <img><|image_2|></img>\n\nInput: <img><|image_3|></img>\nOutput: ",
-            "./imgs/test_cases/icl1.jpg",
-            "./imgs/test_cases/icl2.jpg",
-            "./imgs/test_cases/icl3.jpg",
-            224,
-            224,
-            2.5,
-            1.6,
-            1,
-            768,
-            False,
-            False,
-        ],
+            [   "./imgs/test_cases/icl1.jpg",
+                "./imgs/test_cases/icl2.jpg",
+                "./imgs/test_cases/icl3.jpg",
+                224,
+                224,
+                1.6,
+                1,
+                768,
+                False,
+                False,
+                2.5
+            ],
     ]
     return case
 
-def run_for_examples(text, img1, img2, img3, height, width, guidance_scale, img_guidance_scale, seed, max_input_image_size, randomize_seed, use_input_image_size_as_output):    
+def run_for_examples(img1, img2, img3, height, width, img_guidance_scale, seed, max_input_image_size, randomize_seed,
+                     use_input_image_size_as_output, guidance_scale==3):    
     # Check the internal configuration of the function
     inference_steps = 50
     separate_cfg_infer = True
     offload_model = False
-    
-    return generate_image(text, img1, img2, img3, height, width, guidance_scale, img_guidance_scale, inference_steps, seed,
-                          separate_cfg_infer, offload_model, use_input_image_size_as_output, max_input_image_size, randomize_seed)
+
+    text = "According to the following examples, generate an output for the input.\nInput: <img><|image_1|></img>\nOutput: <img><|image_2|></img>\n\nInput: <img><|image_3|></img>\nOutput:"
+    return generate_image(text, 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)
 
 description = """
 This is a Virtual Try-On Platform.