custom ip

app.py

@@ -1,3 +1,5 @@
+import argparse
+
 import gradio as gr
 import numpy as np
 import os
@@ -12,6 +14,9 @@ from src.datasets import get_dataloader, collate_with_fragment_edges, parse_mole
 from src.lightning import DDPM
 from src.linker_size_lightning import SizeClassifier
 
+parser = argparse.ArgumentParser()
+parser.add_argument(name='ip', type=str, default=None)
+args = parser.parse_args()
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 os.makedirs("results", exist_ok=True)
@@ -176,4 +181,4 @@ with demo:
         outputs=[visualization, output_files],
     )
 
-demo.launch(share=True)
+demo.launch(server_name=args.ip)