Update app.py

app.py

@@ -1,24 +1,26 @@
 import gradio as gr
+import os
 from huggingface_hub import InferenceClient
 
-# Укажи свою модель автодополнения кода (можно заменить на другую)
+# Получаем токен из переменной окружения
+HF_TOKEN = os.getenv("HF_TOKEN")
+
+# Клиент Hugging Face для выбранной модели
 client = InferenceClient(
-    model="deepseek-ai/deepseek-coder-1.3b-base",  # или starcoder2, codegemma, etc.
-    token=HF_TOKKEN  # или укажи свой токен явно: token="hf_xxxxxxxxxxxxxxxxx"
+    model="deepseek-ai/deepseek-coder-1.3b-base",  # Можешь заменить на starcoder2 или codegemma
+    token=HF_TOKEN
 )
 
-# Функция генерации кода с потоковым выводом
+# Функция генерации без stream (deepseek не поддерживает stream=True)
 def complete_code(prompt, max_tokens, temperature, top_p):
-    response = ""
-    for token in client.text_generation(
+    response = client.text_generation(
         prompt,
         max_new_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,
-        stream=True,
-    ):
-        response += token  # token — это уже str
-        yield response
+        stream=False  # ⚠️ обязательно False для этой модели
+    )
+    return response
 
 # Интерфейс Gradio
 demo = gr.Interface(