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)
|