thuzhaowang commited on
Commit
ff723b2
1 Parent(s): 7ad39d9
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -16,6 +16,7 @@ from core.utils.math_utils import unnormalize_params
16
  from huggingface_hub import hf_hub_download
17
 
18
  # Setup PyTorch:
 
19
  device = torch.device('cuda')
20
 
21
  # Define the cache directory for model files
@@ -253,5 +254,5 @@ with gr.Blocks() as demo:
253
  outputs=[output_model_obj],
254
  )
255
 
256
- demo.queue(max_size=10)
257
- demo.launch(server_name="0.0.0.0", server_port=43839)
 
16
  from huggingface_hub import hf_hub_download
17
 
18
  # Setup PyTorch:
19
+ torch.set_grad_enabled(False)
20
  device = torch.device('cuda')
21
 
22
  # Define the cache directory for model files
 
254
  outputs=[output_model_obj],
255
  )
256
 
257
+ demo.queue()
258
+ demo.launch()