Update
Browse files- app.py +1 -1
- app_inference.py +1 -1
- app_training.py +2 -2
- app_upload.py +1 -1
- 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(
|
|
|
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(
|
|
|
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
|
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(
|
|
|
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(
|
|
|
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'
|
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:
|