Test(FIX1)

app.py

@@ -1,21 +1,55 @@
+import os
 import gradio as gr
-from transformers import pipeline
+from openai import OpenAI
 
-# Либо используй свою локальную fine-tuned модель,
-# либо загружай из HF Hub:
-MODEL_PATH = "f3nsmart/MBTIclassifier"
+# =========================================================
+# 1. Настройки API Hugging Face router
+# =========================================================
+# Токен HF (создаётся в https://huggingface.co/settings/tokens)
+HF_TOKEN = os.getenv("HF_TOKEN")
 
-pipe = pipeline("text-classification", model=MODEL_PATH, top_k=3)
 
-def chat(prompt):
-    result = pipe(prompt, max_new_tokens=100)[0]["generated_text"]
-    return result.strip()
+# OpenAI-совместимый клиент с базой HuggingFace Router
+client = OpenAI(
+    base_url="https://router.huggingface.co/v1",
+    api_key=HF_TOKEN,
+)
 
-with gr.Blocks() as demo:
-    gr.Markdown("### 💬 Fine-tuned Mini-LLM Playground")
-    inp = gr.Textbox(label="Введите вопрос или инструкцию")
+# =========================================================
+# 2. Логика общения с моделью
+# =========================================================
+MODEL_ID = "f3nsmart/MBTIclassifier"
+
+def ask_model(prompt):
+    """
+    Отправляет текст модели и возвращает ответ.
+    """
+    try:
+        # Для классификационной модели лучше использовать completions
+        completion = client.completions.create(
+            model=MODEL_ID,
+            prompt=prompt,
+            max_tokens=100
+        )
+        return completion.choices[0].text.strip()
+    except Exception as e:
+        return f"⚠️ Ошибка запроса к модели:\n{e}"
+
+# =========================================================
+# 3. Интерфейс Gradio
+# =========================================================
+with gr.Blocks(title="MBTI Personality Classifier") as demo:
+    gr.Markdown("## 🧠 MBTI Personality Classifier\n"
+                "Введите свой ответ или описание поведения, и модель определит личностные черты.")
+
+    inp = gr.Textbox(label="Введите фразу / описание", placeholder="Например: I enjoy working with people and organizing events.")
     out = gr.Textbox(label="Ответ модели")
-    btn = gr.Button("Сгенерировать")
-    btn.click(fn=chat, inputs=inp, outputs=out)
 
-demo.launch()
+    btn = gr.Button("Определить тип")
+    btn.click(fn=ask_model, inputs=inp, outputs=out)
+
+# =========================================================
+# 4. Запуск
+# =========================================================
+if __name__ == "__main__":
+    demo.launch()