Update app.py

app.py

@@ -1,160 +1,160 @@
-import os
-import cv2
-import random
-import gradio as gr
-from gradio_client import Client, handle_file
-
-machine_number = 0
-model = os.path.join(os.path.dirname(__file__), "models/eva/Eva_0.png")
-client = Client("HumanAIGC/OutfitAnyone")
-
-MODEL_MAP = {
-    "AI Model Rouyan_0": 'models/rouyan_new/Rouyan_0.png',
-    "AI Model Rouyan_1": 'models/rouyan_new/Rouyan_1.png',
-    "AI Model Rouyan_2": 'models/rouyan_new/Rouyan_2.png',
-    "AI Model Eva_0": 'models/eva/Eva_0.png',
-    "AI Model Eva_1": 'models/eva/Eva_1.png',
-    "AI Model Simon_0": 'models/simon_online/Simon_0.png',
-    "AI Model Simon_1": 'models/simon_online/Simon_1.png',
-    "AI Model Xuanxuan_0": 'models/xiaoxuan_online/Xuanxuan_0.png',
-    "AI Model Xuanxuan_1": 'models/xiaoxuan_online/Xuanxuan_1.png',
-    "AI Model Xuanxuan_2": 'models/xiaoxuan_online/Xuanxuan_2.png',
-    "AI Model Yaqi_0": 'models/yaqi/Yaqi_0.png',
-    "AI Model Yaqi_1": 'models/yaqi/Yaqi_1.png',
-    "AI Model Yaqi_2": 'models/yaqi/Yaqi_2.png',
-    "AI Model Yaqi_3": 'models/yaqi/Yaqi_3.png',
-    "AI Model Yifeng_0": 'models/yifeng_online/Yifeng_0.png',
-    "AI Model Yifeng_1": 'models/yifeng_online/Yifeng_1.png',
-    "AI Model Yifeng_2": 'models/yifeng_online/Yifeng_2.png',
-    "AI Model Yifeng_3": 'models/yifeng_online/Yifeng_3.png',
-}
-
-
-def add_waterprint(img):
-    h, w, _ = img.shape
-    img = cv2.putText(img, 'Powered by Infomerica', (int(0.3 * w), h - 20), cv2.FONT_HERSHEY_PLAIN, 2,
-                      (128, 128, 128), 2, cv2.LINE_AA)
-
-    return img
-
-
-def get_tryon_result(model_name, garment1, garment2, seed=1234):
-    # _model = "AI Model " + model_name.split("\\")[-1].split(".")[0]  # windows
-    _model = "AI Model " + model_name.split("/")[-1].split(".")[0]  # linux
-    print("Use Model:", _model)
-    seed = random.randint(0, 1222222222)
-    garment1 = handle_file(garment1) if garment1 else None
-    garment2 = handle_file(garment2) if garment2 else None
-    result = client.predict(
-        handle_file(model_name),
-        garment1,
-        garment2,
-        api_name="/get_tryon_result",
-        fn_index=seed
-    )
-    final_img = remove_watermark2(result)
-    return final_img
-
-
-def remove_watermark2(path):
-    img = cv2.imread(path)
-    img_ = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-    h, w, _ = img.shape
-
-    y_start = max(h - 45, 0)
-    y_end = h
-    x_start = max(int(0.3 * w), 0)
-    x_end = w
-
-    img_[y_start:y_end, x_start:x_end, :] = [255, 255, 255]
-
-    return img_
-
-
-with gr.Blocks(css=".output-image, .input-image, .image-preview {height: 400px !important} ") as demo:
-    # gr.Markdown("# Outfit Anyone v0.9")
-    gr.HTML(
-        """
-        
-        </div>
-        </div>
-        """)
-    with gr.Row():
-        with gr.Column():
-            init_image = gr.Image(sources='clipboard', type="filepath", label="model", value=model)
-            example = gr.Examples(inputs=init_image,
-                                  examples_per_page=4,
-                                  examples=[os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_0')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_2')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_0')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_1')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_1')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_0')),
-                                            os.path.join(os.path.dirname(__file__),
-                                                         MODEL_MAP.get('AI Model Xuanxuan_0')),
-                                            os.path.join(os.path.dirname(__file__),
-                                                         MODEL_MAP.get('AI Model Xuanxuan_2')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_1')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_0')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_3')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_1')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_2')),
-                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_0')),
-                                            ])
-        with gr.Column():
-            gr.HTML(
-                """
-               
-                """)
-            with gr.Row():
-                garment_top = gr.Image(sources='upload', type="filepath", label="top garment")
-                example_top = gr.Examples(inputs=garment_top,
-                                          examples_per_page=5,
-                                          examples=[os.path.join(os.path.dirname(__file__), "garments/top222.JPG"),
-                                                    os.path.join(os.path.dirname(__file__), "garments/top5.png"),
-                                                    os.path.join(os.path.dirname(__file__), "garments/top333.png"),
-                                                    os.path.join(os.path.dirname(__file__), "garments/dress1.png"),
-                                                    os.path.join(os.path.dirname(__file__), "garments/dress2.png"),
-                                                    ])
-                garment_down = gr.Image(sources='upload', type="filepath", label="lower garment")
-                example_down = gr.Examples(inputs=garment_down,
-                                           examples_per_page=5,
-                                           examples=[os.path.join(os.path.dirname(__file__), "garments/bottom1.png"),
-                                                     os.path.join(os.path.dirname(__file__), "garments/bottom2.PNG"),
-                                                     os.path.join(os.path.dirname(__file__), "garments/bottom3.JPG"),
-                                                     os.path.join(os.path.dirname(__file__), "garments/bottom4.PNG"),
-                                                     os.path.join(os.path.dirname(__file__), "garments/bottom5.png"),
-                                                     ])
-
-            run_button = gr.Button(value="Run")
-        with gr.Column():
-            gallery = gr.Image()
-
-            run_button.click(fn=get_tryon_result,
-                             inputs=[
-                                 init_image,
-                                 garment_top,
-                                 garment_down,
-                             ],
-                             outputs=[gallery],
-                             concurrency_limit=2)
-
-    # Examples
-    gr.Markdown("## Examples")
-    with gr.Row():
-        reference_image1 = gr.Image(label="model", scale=1, value="examples/basemodel.png")
-        reference_image2 = gr.Image(label="garment", scale=1, value="examples/garment1.jpg")
-        reference_image3 = gr.Image(label="result", scale=1, value="examples/result1.png")
-    gr.Examples(
-        examples=[
-            ["examples/basemodel.png", "examples/garment1.png", "examples/result1.png"],
-            ["examples/basemodel.png", "examples/garment2.png", "examples/result2.png"],
-            ["examples/basemodel.png", "examples/garment3.png", "examples/result3.png"],
-        ],
-        inputs=[reference_image1, reference_image2, reference_image3],
-        label=None,
-    )
-
-if __name__ == "__main__":
-    demo.queue(max_size=10)
-    demo.launch(share=False, server_name='127.0.0.1', server_port=6006)
+import os
+import cv2
+import random
+import gradio as gr
+from gradio_client import Client, handle_file
+
+machine_number = 0
+model = os.path.join(os.path.dirname(__file__), "models/eva/Eva_0.png")
+client = Client("HumanAIGC/OutfitAnyone")
+
+MODEL_MAP = {
+    "AI Model Rouyan_0": 'models/rouyan_new/Rouyan_0.png',
+    "AI Model Rouyan_1": 'models/rouyan_new/Rouyan_1.png',
+    "AI Model Rouyan_2": 'models/rouyan_new/Rouyan_2.png',
+    "AI Model Eva_0": 'models/eva/Eva_0.png',
+    "AI Model Eva_1": 'models/eva/Eva_1.png',
+    "AI Model Simon_0": 'models/simon_online/Simon_0.png',
+    "AI Model Simon_1": 'models/simon_online/Simon_1.png',
+    "AI Model Xuanxuan_0": 'models/xiaoxuan_online/Xuanxuan_0.png',
+    "AI Model Xuanxuan_1": 'models/xiaoxuan_online/Xuanxuan_1.png',
+    "AI Model Xuanxuan_2": 'models/xiaoxuan_online/Xuanxuan_2.png',
+    "AI Model Yaqi_0": 'models/yaqi/Yaqi_0.png',
+    "AI Model Yaqi_1": 'models/yaqi/Yaqi_1.png',
+    "AI Model Yaqi_2": 'models/yaqi/Yaqi_2.png',
+    "AI Model Yaqi_3": 'models/yaqi/Yaqi_3.png',
+    "AI Model Yifeng_0": 'models/yifeng_online/Yifeng_0.png',
+    "AI Model Yifeng_1": 'models/yifeng_online/Yifeng_1.png',
+    "AI Model Yifeng_2": 'models/yifeng_online/Yifeng_2.png',
+    "AI Model Yifeng_3": 'models/yifeng_online/Yifeng_3.png',
+}
+
+
+def add_waterprint(img):
+    h, w, _ = img.shape
+    img = cv2.putText(img, 'Powered by Infomerica', (int(0.3 * w), h - 20), cv2.FONT_HERSHEY_PLAIN, 2,
+                      (128, 128, 128), 2, cv2.LINE_AA)
+
+    return img
+
+
+def get_tryon_result(model_name, garment1, garment2, seed=1234):
+    # _model = "AI Model " + model_name.split("\\")[-1].split(".")[0]  # windows
+    _model = "AI Model " + model_name.split("/")[-1].split(".")[0]  # linux
+    print("Use Model:", _model)
+    seed = random.randint(0, 1222222222)
+    garment1 = handle_file(garment1) if garment1 else None
+    garment2 = handle_file(garment2) if garment2 else None
+    result = client.predict(
+        handle_file(model_name),
+        garment1,
+        garment2,
+        api_name="/get_tryon_result",
+        fn_index=seed
+    )
+    final_img = remove_watermark2(result)
+    return final_img
+
+
+def remove_watermark2(path):
+    img = cv2.imread(path)
+    img_ = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    h, w, _ = img.shape
+
+    y_start = max(h - 45, 0)
+    y_end = h
+    x_start = max(int(0.3 * w), 0)
+    x_end = w
+
+    img_[y_start:y_end, x_start:x_end, :] = [255, 255, 255]
+
+    return img_
+
+
+with gr.Blocks(css=".output-image, .input-image, .image-preview {height: 400px !important} ") as demo:
+    # gr.Markdown("# Outfit Anyone v0.9")
+    gr.HTML(
+        """
+        
+        </div>
+        </div>
+        """)
+    with gr.Row():
+        with gr.Column():
+            init_image = gr.Image(sources='clipboard', type="filepath", label="model", value=model)
+            example = gr.Examples(inputs=init_image,
+                                  examples_per_page=4,
+                                  examples=[os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_0')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_2')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_0')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_1')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Eva_1')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Simon_0')),
+                                            os.path.join(os.path.dirname(__file__),
+                                                         MODEL_MAP.get('AI Model Xuanxuan_0')),
+                                            os.path.join(os.path.dirname(__file__),
+                                                         MODEL_MAP.get('AI Model Xuanxuan_2')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_1')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_0')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_3')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Rouyan_1')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yifeng_2')),
+                                            os.path.join(os.path.dirname(__file__), MODEL_MAP.get('AI Model Yaqi_0')),
+                                            ])
+        with gr.Column():
+            gr.HTML(
+                """
+               
+                """)
+            with gr.Row():
+                garment_top = gr.Image(sources='upload', type="filepath", label="top garment")
+                example_top = gr.Examples(inputs=garment_top,
+                                          examples_per_page=5,
+                                          examples=[os.path.join(os.path.dirname(__file__), "garments/top222.JPG"),
+                                                    os.path.join(os.path.dirname(__file__), "garments/top5.png"),
+                                                    os.path.join(os.path.dirname(__file__), "garments/top333.png"),
+                                                    os.path.join(os.path.dirname(__file__), "garments/dress1.png"),
+                                                    os.path.join(os.path.dirname(__file__), "garments/dress2.png"),
+                                                    ])
+                garment_down = gr.Image(sources='upload', type="filepath", label="lower garment")
+                example_down = gr.Examples(inputs=garment_down,
+                                           examples_per_page=5,
+                                           examples=[os.path.join(os.path.dirname(__file__), "garments/bottom1.png"),
+                                                     os.path.join(os.path.dirname(__file__), "garments/bottom2.PNG"),
+                                                     os.path.join(os.path.dirname(__file__), "garments/bottom3.JPG"),
+                                                     os.path.join(os.path.dirname(__file__), "garments/bottom4.PNG"),
+                                                     os.path.join(os.path.dirname(__file__), "garments/bottom5.png"),
+                                                     ])
+
+            run_button = gr.Button(value="Run")
+        with gr.Column():
+            gallery = gr.Image()
+
+            run_button.click(fn=get_tryon_result,
+                             inputs=[
+                                 init_image,
+                                 garment_top,
+                                 garment_down,
+                             ],
+                             outputs=[gallery],
+                             concurrency_limit=2)
+
+    # Examples
+    gr.Markdown("## Examples")
+    with gr.Row():
+        reference_image1 = gr.Image(label="model", scale=1, value="examples/basemodel.png")
+        reference_image2 = gr.Image(label="garment", scale=1, value="examples/garment1.jpg")
+        reference_image3 = gr.Image(label="result", scale=1, value="examples/result1.png")
+    gr.Examples(
+        examples=[
+            ["examples/basemodel.png", "examples/garment1.png", "examples/result1.png"],
+            ["examples/basemodel.png", "examples/garment2.png", "examples/result2.png"],
+            ["examples/basemodel.png", "examples/garment3.png", "examples/result3.png"],
+        ],
+        inputs=[reference_image1, reference_image2, reference_image3],
+        label=None,
+    )
+
+if __name__ == "__main__":
+    demo.queue(max_size=10)
+    demo.launch()