deepkyu commited on
Commit
690f7ad
1 Parent(s): 804cca3

fix variable name

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -8,8 +8,8 @@ from demo import SdmCompressionDemo
8
 
9
  dest_path_config = Path('checkpoints/BK-SDM-Small_iter50000/unet/config.json')
10
  dest_path_torch_ckpt = Path('checkpoints/BK-SDM-Small_iter50000/unet/diffusion_pytorch_model.bin')
11
- BK_SDM_CONFIG_URL: str = os.getenv('CHECKPOINT_CONFIG', None)
12
- BK_SDM_TORCH_CKPT_URL: str = os.getenv('CHECKPOINT_PYTORCH_BIN', None)
13
  assert BK_SDM_CONFIG_URL is not None
14
  assert BK_SDM_TORCH_CKPT_URL is not None
15
 
@@ -80,5 +80,4 @@ if __name__ == "__main__":
80
 
81
  demo.queue(concurrency_count=1)
82
  # demo.launch()
83
- demo.launch(share=True, auth=("test", "testasdf@@19"))
84
-
 
8
 
9
  dest_path_config = Path('checkpoints/BK-SDM-Small_iter50000/unet/config.json')
10
  dest_path_torch_ckpt = Path('checkpoints/BK-SDM-Small_iter50000/unet/diffusion_pytorch_model.bin')
11
+ BK_SDM_CONFIG_URL: str = os.getenv('BK_SDM_CONFIG_URL', None)
12
+ BK_SDM_TORCH_CKPT_URL: str = os.getenv('BK_SDM_TORCH_CKPT_URL', None)
13
  assert BK_SDM_CONFIG_URL is not None
14
  assert BK_SDM_TORCH_CKPT_URL is not None
15
 
 
80
 
81
  demo.queue(concurrency_count=1)
82
  # demo.launch()
83
+ demo.launch(server_name="0.0.0.0", server_port=7861)