push

app.py

@@ -1,10 +1,15 @@
 import os
 import gradio as gr
 from huggingface_hub import InferenceClient
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 
-# Используем токен из secrets
-client = InferenceClient(os.getenv("HUGGINGFACE_TOKEN"))
+# Убедись, что токен загружен из переменной окружения или напрямую
+HF_TOKEN = os.getenv("HUGGINGFACE_TOKEN")  # Загрузка токена из .env
 
+# Инициализация клиента
+client = InferenceClient("sambanovasystems/SambaLingo-Russian-Chat", token=HF_TOKEN)
+
+# Функция для ответа на сообщения
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -25,29 +30,24 @@ def respond(
 
     response = ""
 
-    try:
-        # Указываем модель
-        for message in client.chat_completion(
-            model="sambanovasystems/SambaLingo-Russian-Chat",
-            messages=messages,
-            max_tokens=max_tokens,
-            stream=True,
-            temperature=temperature,
-            top_p=top_p,
-        ):
-            token = message.choices[0].delta.content
-            response += token
-            yield response
-    except Exception as e:
-        print(f"Error: {e}")
-        yield "Произошла ошибка при обработке вашего запроса."
-
+    for message in client.chat_completion(
+        messages,
+        max_tokens=max_tokens,
+        stream=True,
+        temperature=temperature,
+        top_p=top_p,
+    ):
+        token = message.choices[0].delta.content
+        response += token
+        yield response
+
+# Настройка интерфейса Gradio
 demo = gr.ChatInterface(
     respond,
     additional_inputs=[
-        gr.Textbox(value="Ты дружелюбный чат-бот.", label="Системное сообщение"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Максимум новых токенов"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Температура"),
+        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
         gr.Slider(
             minimum=0.1,
             maximum=1.0,