Update app.py

app.py

@@ -4,16 +4,19 @@ from transformers import pipeline
 generator = pipeline(
     "text-generation",
     model="deepseek-ai/deepseek-coder-1.3b-instruct",
+    device_map="auto".
     clean_up_tokenization_spaces=True
 )
 
 SYSTEM_PROMPT = """
-You are an expert Python developer.
-
-Always:
-- write clean Python code
-- add comments when useful
-- avoid unnecessary explanations
+You are a senior Python engineer.
+
+When answering:
+- Return clean Python code
+- Follow best practices
+- Keep functions simple
+- Add comments if useful
+- Avoid unnecessary explanations
 """
 
 def chat(message, history):
@@ -31,7 +34,9 @@ def chat(message, history):
     # Clean tokenizer artifacts
     output = output.replace("Ċ", "\n").replace("Ġ", " ")
 
-    return output.split("Assistant:")[-1]
+    code = output.split("Assistant:")[-1]
+
+    return f"```python\n{code}\n```"
 
 
 demo = gr.ChatInterface(