update

app.py

@@ -29,16 +29,16 @@ from inference import inference_fn
 # from uploader import upload
 
 
-def parse_args() -> argparse.Namespace:
-    parser = argparse.ArgumentParser()
-    parser.add_argument('--device', type=str, default='cpu')
-    parser.add_argument('--theme', type=str)
-    parser.add_argument('--share', action='store_true')
-    parser.add_argument('--port', type=int)
-    parser.add_argument('--disable-queue',
-                        dest='enable_queue',
-                        action='store_false')
-    return parser.parse_args()
+# def parse_args() -> argparse.Namespace:
+#     parser = argparse.ArgumentParser()
+#     parser.add_argument('--device', type=str, default='cpu')
+#     parser.add_argument('--theme', type=str)
+#     parser.add_argument('--share', action='store_true')
+#     parser.add_argument('--port', type=int)
+#     parser.add_argument('--disable-queue',
+#                         dest='enable_queue',
+#                         action='store_false')
+#     return parser.parse_args()
 
 
 TITLE = '# ReVersion'
@@ -176,9 +176,13 @@ def create_inference_demo(func: inference_fn) -> gr.Blocks:
     return demo
 
 
-args = parse_args()
-args.device = 'cuda' if torch.cuda.is_available() else 'cpu'
-print('*** Now using %s.'%(args.device))
+# args = parse_args()
+# args.device = 'cuda' if torch.cuda.is_available() else 'cpu'
+# print('*** Now using %s.'%(args.device))
+if torch.cuda.is_available():
+    print('*** Now using %s.'%('cuda'))
+else:
+    print('*** Now using %s.'%('cpu'))
 
 with gr.Blocks(css='style.css') as demo:
     # if os.getenv('IS_SHARED_UI'):
@@ -194,10 +198,12 @@ with gr.Blocks(css='style.css') as demo:
         with gr.TabItem('Relation-Specific Text-to-Image Generation'):
             create_inference_demo(inference_fn)
 
-demo.launch(
-    enable_queue=args.enable_queue,
-    server_port=args.port,
-    share=args.share
-)
+demo.queue(default_enabled=False).launch(share=False)
+
+# demo.launch(
+#     enable_queue=args.enable_queue,
+#     server_port=args.port,
+#     share=args.share
+# )
 # demo.queue(default_enabled=False).launch(server_port=args.port, share=args.share)