app.py

# Use a pipeline as a high-level helper
from transformers import pipeline
import streamlit as st

st.title("AI Chatbot")

st.write("Ask the AI a question and get an answer!")

user_question = st.chat_input("Enter your question:")

messages = [
    {"role": "user", "content": user_question},
]
pipe = pipeline("text-generation", model="deepseek-ai/DeepSeek-R1-Zero", trust_remote_code=True)


st.write(pipe(messages))