Update app.py

app.py

@@ -4,7 +4,7 @@ from multit2i import (load_models, infer_fn, infer_rand_fn, save_gallery,
     change_model, warm_model, get_model_info_md, loaded_models,
     get_positive_prefix, get_positive_suffix, get_negative_prefix, get_negative_suffix,
     get_recom_prompt_type, set_recom_prompt_preset, get_tag_type, randomize_seed, translate_to_en)
-
+from huggingface_hub import login
 max_images = 8
 MAX_SEED = 2**32-1
 load_models(models)
@@ -15,6 +15,20 @@ css = """
 .gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
 """
 
+
+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)
 with gr.Blocks(theme="NoCrypt/miku@>=1.2.2", fill_width=True, css=css) as demo:
     with gr.Tab("Image Generator"):
         with gr.Row():