Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| generator = pipeline( | |
| "text-generation", | |
| model="deepseek-ai/deepseek-coder-1.3b-instruct" | |
| 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 | |
| """ | |
| def chat(message, history): | |
| prompt = SYSTEM_PROMPT + "\nUser: " + message + "\nAssistant:" | |
| result = generator( | |
| prompt, | |
| max_new_tokens=200, | |
| temperature=0.2 | |
| ) | |
| output = result[0]["generated_text"] | |
| # Clean tokenizer artifacts | |
| output = output.replace("Ċ", "\n").replace("Ġ", " ") | |
| return output.split("Assistant:")[-1] | |
| demo = gr.ChatInterface( | |
| fn=chat, | |
| title="Python Coding Assistant", | |
| description="Ask Python coding questions" | |
| ) | |
| demo.launch() |