Update app.py

app.py

@@ -4,9 +4,10 @@ from typing import List, Dict, Any, Tuple
 import torch
 
 # CPU-модели (только одна маленькая модель для экономии памяти)
+# Исправлено: используем словарь вместо множества
 MODELS = {
-    "Qwen/Qwen2.5-0.5B-Instruct",
-    "Qwen/Qwen2.5-1.5B-Instruct",
+    "Qwen2.5-0.5B": "Qwen/Qwen2.5-0.5B-Instruct",
+    "Qwen2.5-1.5B": "Qwen/Qwen2.5-1.5B-Instruct",
 }
 
 def load_model(model_key: str):
@@ -76,7 +77,8 @@ with gr.Blocks(title="🚀 Локальный HF Чат (на слабом CPU!)
     gr.Markdown("# Локальный Inference (без API!)\n**Маленькие модели** — 1-3 сек CPU. Большие думают ооочень долго. Нет limits/token. В качестве примера.")
     
     with gr.Row():
-        model_dropdown = gr.Dropdown(choices=list(MODELS.keys()), value="gpt2", label="🧠 Модель")
+        # Исправлено: первое значение должно быть ключом из словаря MODELS
+        model_dropdown = gr.Dropdown(choices=list(MODELS.keys()), value="Qwen2.5-0.5B", label="🧠 Модель")
         system_prompt = gr.Textbox(label="📝 System", placeholder="Ты весёлый ИИ.", lines=2)
     
     chatbot = gr.Chatbot(height=500, label="Чат")  # ← без type