MrDrmm commited on
Commit
ae4bb43
·
verified ·
1 Parent(s): af35444

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +36 -0
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)