Update multit2i.py

multit2i.py

@@ -11,6 +11,18 @@ server_timeout = 2600
 inference_timeout = 1400
 
 
+def set_hf_token(token):
+    os.environ["HF_TOKEN"] = token
+    return f"Токен установлен: {token[:5]}...{token[-5:]}"
+
+def generate_image(prompt, width, height):
+    if "HF_TOKEN" not in os.environ or not os.environ["HF_TOKEN"]:
+        return "Ошибка: Токен Hugging Face не установлен!"
+    
+    model = gr.load("models/digiplay/rRealism_v1.0_riiwa", hf_token=os.environ["HF_TOKEN"])
+    result = model.predict(prompt=prompt, resolution=f"{width}x{height}")
+    return result
+
 lock = RLock()
 loaded_models = {}
 model_info_dict = {}