File size: 266 Bytes
492ddf3 |
1 2 3 4 5 6 7 8 9 |
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)) |