Update multit2i.py

multit2i.py

@@ -4,6 +4,21 @@ from threading import RLock
 from pathlib import Path
 from huggingface_hub import InferenceClient
 import os
+from huggingface_hub import login
+
+def set_token(token):
+    global HF_TOKEN
+    HF_TOKEN = token
+    os.environ["HF_TOKEN"] = token  # Сохраняем в переменной окружения
+    login(HF_TOKEN)
+    return "Токен установлен!"
+
+with gr.Blocks() as demo:
+    hf_token_input = gr.Textbox(label="Введите HF_TOKEN", type="password")
+    submit_button = gr.Button("Сохранить")
+    output_text = gr.Textbox(label="Статус")
+
+    submit_button.click(set_token, inputs=hf_token_input, outputs=output_text)
 
 
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.