Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 import torch
 
-# 1. ЗАГРУЗКА МОЗГА (Он уже скачан в кэш, так что будет быстро!)
+# 1. ЗАГРУЗКА МОЗГА (Он уже в кэше, запустится быстро)
 model_name = "HuggingFaceTB/SmolLM-135M-Instruct"
 device = 0 if torch.cuda.is_available() else -1
 generator = pipeline("text-generation", model=model_name, device=device)
@@ -31,25 +31,27 @@ body {
 
 # 3. ЛОГИКА УМНОГО ЧАТА
 def axis_chat(message, history):
-    # Специальный формат запроса для этой модели (чтобы она не тупила)
+    # Промпт для общения
     prompt = f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
     # Генерация ответа
     output = generator(prompt, max_new_tokens=150, do_sample=True, temperature=0.7)
     
-    # Очищаем ответ от системных тегов
+    # Очистка текста от тегов
     full_text = output[0]['generated_text']
-    answer = full_text.split("<|im_start|>assistant\n")[-1].replace("<|im_end|>", "").strip()
+    if "<|im_start|>assistant\n" in full_text:
+        answer = full_text.split("<|im_start|>assistant\n")[-1].replace("<|im_end|>", "").strip()
+    else:
+        answer = full_text
     return answer
 
 # 4. СОЗДАНИЕ ИНТЕРФЕЙСА
 with gr.Blocks() as demo:
     gr.HTML("<div class='quality-text'>PROJECT AXIS: EVOLVED AI</div>")
-    gr.ChatInterface(
-        fn=axis_chat,
-        type="messages" # Новый стандарт Gradio 6
-    )
+    
+    # Простое и надежное создание чата
+    gr.ChatInterface(fn=axis_chat)
 
 if __name__ == "__main__":
-    # В Gradio 6.0 CSS передается именно сюда!
+    # Передаем CSS в launch, как просил сервер в прошлый раз
     demo.launch(css=css)