Update app.py

app.py

@@ -659,24 +659,13 @@ if __name__ == "__main__":
         login(token=api_key)
         logger.info("Logged in to Hugging Face Hub")
 
-    # Launch Gradio app in a subprocess to avoid CUDA initialization in the main process
-    from torch.multiprocessing import Process
-
-    def launch_demo():
-        try:
-            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
-            )
-        except Exception as e:
-            logger.error(f"Error launching demo: {e}")
-            sys.exit(1)
 
-    p = Process(target=launch_demo)
-    p.start()
-    p.join()
+    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
+    )