Update app.py

app.py

@@ -632,40 +632,39 @@ def build_demo(embed_mode, cur_dir=None, concurrency_count=10):
     return demo
 
 
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser()
-    parser.add_argument("--host", type=str, default="0.0.0.0")
-    parser.add_argument("--port", type=int)
-    parser.add_argument("--concurrency-count", type=int, default=5)
-    parser.add_argument("--share", action="store_true")
-    parser.add_argument("--moderate", action="store_true")
-    parser.add_argument("--embed", action="store_true")
-    args = parser.parse_args()
-
-    # Create log directory if it doesn't exist
-    os.makedirs(LOGDIR, exist_ok=True)
-
-    # GPU Check
-    if torch.cuda.is_available():
-        logger.info(f"CUDA available with {torch.cuda.device_count()} devices")
-    else:
-        logger.warning("CUDA not available! Models will run on CPU which may be very slow.")
-
-    # Hugging Face token handling
-    api_key = os.getenv("token")
-    if api_key:
-        from huggingface_hub import login
-
-        login(token=api_key)
-        logger.info("Logged in to Hugging Face Hub")
-
-
-    demo = build_demo(embed_mode=args.embed, cur_dir='./', concurrency_count=args.concurrency_count)
-    demo.queue(
-        status_update_rate=10,
-        api_open=False
-    ).launch(
-        server_name=args.host,
-        server_port=args.port,
-        share=args.share
-    )
+parser = argparse.ArgumentParser()
+parser.add_argument("--host", type=str, default="0.0.0.0")
+parser.add_argument("--port", type=int)
+parser.add_argument("--concurrency-count", type=int, default=5)
+parser.add_argument("--share", action="store_true")
+parser.add_argument("--moderate", action="store_true")
+parser.add_argument("--embed", action="store_true")
+args = parser.parse_args()
+
+# Create log directory if it doesn't exist
+os.makedirs(LOGDIR, exist_ok=True)
+
+# GPU Check
+if torch.cuda.is_available():
+    logger.info(f"CUDA available with {torch.cuda.device_count()} devices")
+else:
+    logger.warning("CUDA not available! Models will run on CPU which may be very slow.")
+
+# Hugging Face token handling
+api_key = os.getenv("token")
+if api_key:
+    from huggingface_hub import login
+
+    login(token=api_key)
+    logger.info("Logged in to Hugging Face Hub")
+
+
+demo = build_demo(embed_mode=args.embed, cur_dir='./', concurrency_count=args.concurrency_count)
+demo.queue(
+    status_update_rate=10,
+    api_open=False
+).launch(
+    server_name=args.host,
+    server_port=args.port,
+    share=args.share
+)