Update app.py

app.py

@@ -2,54 +2,54 @@ 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)
 
-# 2. ДИЗАЙН (Исправляем видимость текста и добавляем твой градиент)
+# 2. ДИЗАЙН (Твой градиент)
 css = """
-footer {visibility: hidden}
 body {
     background: linear-gradient(135deg, #FF00CC 0%, #9932CC 40%, #0000FF 70%, #00FFFF 100%) !important;
     background-attachment: fixed;
 }
-.gr-container {
-    background: rgba(0, 0, 0, 0.7) !important; /* Темный фон для чата, чтобы текст сиял */
-    border-radius: 15px;
+.gradio-container {
+    background: rgba(0, 0, 0, 0.7) !important;
+    border-radius: 20px;
     border: 1px solid #00FFFF;
+    color: white !important;
 }
 .quality-text {
-    color: #00FFFF; /* Яркий бирюзовый, будет отлично виден */
-    text-shadow: 0 0 10px #00FFFF;
+    color: #00FFFF;
+    text-shadow: 0 0 15px #00FFFF;
     font-weight: bold;
-    font-size: 30px;
+    font-size: 32px;
     text-align: center;
-    padding: 10px;
+    margin: 20px;
 }
-p, span, label { color: white !important; font-size: 16px; }
 """
 
-# 3. ЛОГИКА ОБЩЕНИЯ
+# 3. ЛОГИКА УМНОГО ЧАТА
 def axis_chat(message, history):
-    # Формируем запрос для ИИ
-    prompt = f"User: {message}\nAssistant:"
+    # Специальный формат запроса для этой модели (чтобы она не тупила)
+    prompt = f"<|im_start|>user\n{message}<|im_end|>\n<|im_start|>assistant\n"
     
-    # Генерируем ответ
-    output = generator(prompt, max_new_tokens=50, do_sample=True, temperature=0.7)
-    answer = output[0]['generated_text'].split("Assistant:")[-1].strip()
+    # Генерация ответа
+    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()
     return answer
 
-# 4. ЗАПУСК ТЕРМИНАЛА
-with gr.Blocks(css=css) as demo:
+# 4. СОЗДАНИЕ ИНТЕРФЕЙСА
+with gr.Blocks() as demo:
     gr.HTML("<div class='quality-text'>PROJECT AXIS: EVOLVED AI</div>")
-    
     gr.ChatInterface(
         fn=axis_chat,
-        description="Система прошла мгновенное обучение. Мозги: SmolLM-Instruct."
+        type="messages" # Новый стандарт Gradio 6
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    # В Gradio 6.0 CSS передается именно сюда!
+    demo.launch(css=css)