| import gradio as gr | |
| from transformers import pipeline | |
| def chat_with_model(user_input): | |
| messages = [{"role": "user", "content": user_input}] | |
| pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True) | |
| response = pipe(messages) | |
| return response[0]['generated_text'] | |
| iface = gr.Interface( | |
| fn=chat_with_model, | |
| inputs=gr.Textbox(label="Enter your message"), | |
| outputs=gr.Textbox(label="Model Response"), | |
| title="Chat with DeepSeek-R1", | |
| description="Enter a message to interact with the DeepSeek-R1 model." | |
| ) | |
| iface.launch() | |