1

app.py

@@ -8,91 +8,94 @@ import gradio as gr
 from detect import detect
 from util import get_all_weights
 
-app_introduce = """
-# CycleGAN
-功能：上传本地文件、选择转换风格
-"""
-
-css = """
-# :root{
-# --block-background-fill: #f3f3f3;
-# }
-footer{
-    display:none!important;
-}
-"""
-
-demo = gr.Blocks(
-    # theme=gr.themes.Soft(),
-    css=css
-)
-
 
-def add_img(img):
-    imgs = []
-    for style in get_all_weights():
-        fake_img = detect(img, style=style)
-        imgs.append(fake_img)
-    return imgs
-
-
-def tab1(label="上传单张图片"):
-    default_img_paths = [
-        [Path.cwd().joinpath("./imgs/horse.jpg"), "horse2zebra"],
-        [Path.cwd().joinpath("./imgs/monet.jpg"), "monet2photo"],
-    ]
-
-    with gr.Tab(label):
-        with gr.Row():
-            with gr.Column():
-                img = gr.Image(type="pil", label="选择需要进行风格转换的图片")
-                style = gr.Dropdown(choices=get_all_weights(), label="转换的风格")
-                detect_btn = gr.Button("♻️风格转换")
+def main():
+    app_introduce = """
+    # CycleGAN
+    功能：上传本地文件、选择转换风格
+    """
+
+    css = """
+    # :root{
+    # --block-background-fill: #f3f3f3;
+    # }
+    # footer{
+    #     display:none!important;
+    # }
+    """
+
+    demo = gr.Blocks(
+        # theme=gr.themes.Soft(),
+        css=css
+    )
+
+
+    def add_img(img):
+        imgs = []
+        for style in get_all_weights():
+            fake_img = detect(img, style=style)
+            imgs.append(fake_img)
+        return imgs
+
+
+    def tab1(label="上传单张图片"):
+        default_img_paths = [
+            [Path.cwd().joinpath("./imgs/horse.jpg"), "horse2zebra"],
+            [Path.cwd().joinpath("./imgs/monet.jpg"), "monet2photo"],
+        ]
+
+        with gr.Tab(label):
+            with gr.Row():
+                with gr.Column():
+                    img = gr.Image(type="pil", label="选择需要进行风格转换的图片")
+                    style = gr.Dropdown(choices=get_all_weights(), label="转换的风格")
+                    detect_btn = gr.Button("♻️风格转换")
+                with gr.Column():
+                    out_img = gr.Image(label="风格图")
+            detect_btn.click(fn=detect, inputs=[img, style], outputs=[out_img])
+            gr.Examples(default_img_paths, inputs=[img, style])
+
+
+    def tab2(label="单图多风格试转换"):
+        with gr.Tab(label):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    img = gr.Image(type="pil", label="选择需要进行风格转换的图片")
+                    gr.Markdown("上传一张图片，会将所有风格推理一遍。")
+                    btn = gr.Button("♻️风格转换")
+                with gr.Column(scale=2):
+                    gallery = gr.Gallery(
+                        label="风格图",
+                        elem_id="gallery",
+                    ).style(grid=[3], height="auto")
+            btn.click(fn=add_img, inputs=[img], outputs=gallery)
+
+
+    def tab3(label="参数设置"):
+        from detect import opt
+
+        with gr.Tab(label):
             with gr.Column():
-                out_img = gr.Image(label="风格图")
-        detect_btn.click(fn=detect, inputs=[img, style], outputs=[out_img])
-        gr.Examples(default_img_paths, inputs=[img, style])
-
-
-def tab2(label="单图多风格试转换"):
-    with gr.Tab(label):
-        with gr.Row():
-            with gr.Column(scale=1):
-                img = gr.Image(type="pil", label="选择需要进行风格转换的图片")
-                gr.Markdown("上传一张图片，会将所有风格推理一遍。")
-                btn = gr.Button("♻️风格转换")
-            with gr.Column(scale=2):
-                gallery = gr.Gallery(
-                    label="风格图",
-                    elem_id="gallery",
-                ).style(grid=[3], height="auto")
-        btn.click(fn=add_img, inputs=[img], outputs=gallery)
-
-
-def tab3(label="参数设置"):
-    from detect import opt
-
-    with gr.Tab(label):
-        with gr.Column():
-            for k, v in sorted(vars(opt).items()):
-                if type(v) == bool:
-                    gr.Checkbox(label=k, value=v)
-                elif type(v) == (int or float):
-                    gr.Number(label=k, value=v)
-                elif type(v) == list:
-                    gr.CheckboxGroup(label=k, value=v)
-                else:
-                    gr.Textbox(label=k, value=v)
-
-
-with demo:
-    gr.Markdown(app_introduce)
-    tab1()
-    tab2()
-    tab3()
+                for k, v in sorted(vars(opt).items()):
+                    if type(v) == bool:
+                        gr.Checkbox(label=k, value=v)
+                    elif type(v) == (int or float):
+                        gr.Number(label=k, value=v)
+                    elif type(v) == list:
+                        gr.CheckboxGroup(label=k, value=v)
+                    else:
+                        gr.Textbox(label=k, value=v)
+
+
+    with demo:
+        gr.Markdown(app_introduce)
+        tab1()
+        tab2()
+        tab3()
+    demo.launch(server_name="0.0.0.0", server_port = ARGS.port)
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    main()
     # 如果不以`demo`命名，`gradio app.py`会报错`Error loading ASGI app. Attribute "demo.app" not found in module "app".`
     # 注意gradio库的reload.py的头信息 $ gradio app.py my_demo, to use variable names other than "demo"
     # my_demo 是定义的变量。离谱o(╥﹏╥)o

weights/detect/test_opt.txt

@@ -9,7 +9,7 @@
           display_winsize: 256                           
                     epoch: latest                        
                      eval: False                         
-                  gpu_ids: 0                             
+                  gpu_ids: -1                            
                      help: folder: /imgs/ or file: xx.jpg
                 init_gain: 0.02                          
                 init_type: normal