Update modules/ui.py

Co-authored-by: Víctor Gallego <vicgalle@ucm.es>
This commit is contained in:
MalumaDev 2022-10-15 18:41:46 +02:00 committed by GitHub
parent ad9bc604a8
commit 3f5c3b981e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1190,7 +1190,8 @@ def create_ui(wrap_gradio_gpu_call):
with gr.Column():
create_embedding = gr.Button(value="Create embedding", variant='primary')
with gr.Tab(label="Create images embedding"):
with gr.Tab(label="Create aesthetic images embedding"):
new_embedding_name_ae = gr.Textbox(label="Name")
process_src_ae = gr.Textbox(label='Source directory')
batch_ae = gr.Slider(minimum=1, maximum=1024, step=1, label="Batch size", value=256)