hysts HF staff commited on
Commit
db0cd98
1 Parent(s): 06affa1
Files changed (5) hide show
  1. app.py +1 -1
  2. app_inference.py +1 -1
  3. app_training.py +2 -2
  4. app_upload.py +1 -1
  5. trainer.py +1 -1
app.py CHANGED
@@ -79,4 +79,4 @@ with gr.Blocks(css='style.css') as demo:
79
  if not HF_TOKEN:
80
  show_warning(HF_TOKEN_NOT_SPECIFIED_WARNING)
81
 
82
- demo.queue(max_size=1).launch(share=False)
 
79
  if not HF_TOKEN:
80
  show_warning(HF_TOKEN_NOT_SPECIFIED_WARNING)
81
 
82
+ demo.queue(api_open=False, max_size=1).launch()
app_inference.py CHANGED
@@ -167,4 +167,4 @@ if __name__ == '__main__':
167
  hf_token = os.getenv('HF_TOKEN')
168
  pipe = InferencePipeline(hf_token)
169
  demo = create_inference_demo(pipe, hf_token)
170
- demo.queue(max_size=10).launch(share=False)
 
167
  hf_token = os.getenv('HF_TOKEN')
168
  pipe = InferencePipeline(hf_token)
169
  demo = create_inference_demo(pipe, hf_token)
170
+ demo.queue(api_open=False, max_size=10).launch()
app_training.py CHANGED
@@ -6,7 +6,7 @@ import os
6
 
7
  import gradio as gr
8
 
9
- from constants import MODEL_LIBRARY_ORG_NAME, SAMPLE_MODEL_REPO, UploadTarget
10
  from inference import InferencePipeline
11
  from trainer import Trainer
12
 
@@ -132,4 +132,4 @@ if __name__ == '__main__':
132
  hf_token = os.getenv('HF_TOKEN')
133
  trainer = Trainer(hf_token)
134
  demo = create_training_demo(trainer)
135
- demo.queue(max_size=1).launch(share=False)
 
6
 
7
  import gradio as gr
8
 
9
+ from constants import UploadTarget
10
  from inference import InferencePipeline
11
  from trainer import Trainer
12
 
 
132
  hf_token = os.getenv('HF_TOKEN')
133
  trainer = Trainer(hf_token)
134
  demo = create_training_demo(trainer)
135
+ demo.queue(api_open=False, max_size=1).launch()
app_upload.py CHANGED
@@ -103,4 +103,4 @@ if __name__ == '__main__':
103
 
104
  hf_token = os.getenv('HF_TOKEN')
105
  demo = create_upload_demo(hf_token)
106
- demo.queue(max_size=1).launch(share=False)
 
103
 
104
  hf_token = os.getenv('HF_TOKEN')
105
  demo = create_upload_demo(hf_token)
106
+ demo.queue(api_open=False, max_size=1).launch()
trainer.py CHANGED
@@ -21,7 +21,7 @@ sys.path.append('Tune-A-Video')
21
 
22
  URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/Tune-A-Video-library/share/YjTcaNJmKyeHFpMBioHhzBcTzCYddVErEk'
23
  ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
24
- SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
25
 
26
 
27
  class Trainer:
 
21
 
22
  URL_TO_JOIN_MODEL_LIBRARY_ORG = 'https://huggingface.co/organizations/Tune-A-Video-library/share/YjTcaNJmKyeHFpMBioHhzBcTzCYddVErEk'
23
  ORIGINAL_SPACE_ID = 'Tune-A-Video-library/Tune-A-Video-Training-UI'
24
+ SPACE_ID = os.getenv('SPACE_ID')
25
 
26
 
27
  class Trainer: