Update app.py
Browse files
app.py
CHANGED
@@ -6,6 +6,42 @@ from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
|
|
6 |
get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
|
7 |
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
max_images = 8
|
10 |
MAX_SEED = 2**32-1
|
11 |
load_models(models)
|
|
|
6 |
get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
|
7 |
|
8 |
|
9 |
+
import os
|
10 |
+
|
11 |
+
# Поле для HF_TOKEN
|
12 |
+
hf_token_input = gr.Textbox(label="Введите HF Token", type="password", interactive=True)
|
13 |
+
|
14 |
+
def set_hf_token(hf_token):
|
15 |
+
"""Функция для установки HF_TOKEN и проверки его активации"""
|
16 |
+
if hf_token:
|
17 |
+
os.environ["HF_TOKEN"] = hf_token # Устанавливаем токен в переменные окружения
|
18 |
+
return f"✅ HF_TOKEN установлен!", gr.update(value=hf_token, interactive=True)
|
19 |
+
else:
|
20 |
+
return "❌ Токен не введен!", gr.update(value="", interactive=True)
|
21 |
+
|
22 |
+
# Кнопка подтверждения токена
|
23 |
+
confirm_token = gr.Button("Активировать HF_TOKEN")
|
24 |
+
|
25 |
+
# Поле для отображения статуса
|
26 |
+
token_status = gr.Markdown("🔴 HF_TOKEN не установлен")
|
27 |
+
|
28 |
+
# Подключаем обработчик кнопки
|
29 |
+
confirm_token.click(set_hf_token, inputs=[hf_token_input], outputs=[token_status, hf_token_input])
|
30 |
+
|
31 |
+
# Вставляем эти элементы в твой UI
|
32 |
+
with gr.Blocks(theme="NoCrypt/miku@>=1.2.2") as demo:
|
33 |
+
with gr.Tab("Settings"):
|
34 |
+
gr.Markdown("### Настройки токена Hugging Face")
|
35 |
+
hf_token_input.render() # Поле ввода
|
36 |
+
confirm_token.render() # Кнопка активации
|
37 |
+
token_status.render() # Статус токена
|
38 |
+
|
39 |
+
# Здесь твой код UI (его менять не нужно)
|
40 |
+
# ...
|
41 |
+
|
42 |
+
|
43 |
+
|
44 |
+
|
45 |
max_images = 8
|
46 |
MAX_SEED = 2**32-1
|
47 |
load_models(models)
|