import streamlit as st
from transformers import pipeline

message = st.text_input('message')
messages = [
    {"role": "user", "content": message},
]
pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
st.write(pipe(messages))