modify main function

app.py

@@ -531,17 +531,16 @@ def build_demo(embed_mode, cur_dir=None):
     return demo
 
 
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--share", action="store_true")
-    parser.add_argument("--embed", action="store_true")
-    args = parser.parse_args()
-
-    demo = build_demo(args.embed)
-    demo.queue(
-        api_open=False
-    ).launch(
-        favicon_path="./examples/icon_256.png",
-        allowed_paths=["/"],
-        share=args.share
-    )
+parser = argparse.ArgumentParser()
+parser.add_argument("--share", action="store_true")
+parser.add_argument("--embed", action="store_true")
+args = parser.parse_args()
+
+demo = build_demo(args.embed)
+demo.queue(
+    api_open=False
+).launch(
+    favicon_path="./examples/icon_256.png",
+    allowed_paths=["/"],
+    share=args.share
+)