| import streamlit as st | |
| import os | |
| from transformers import pipeline, Conversation | |
| convo = pipeline("conversational") | |
| def chatbot(): | |
| st.title("ChatBot") | |
| if message := st.chat_input("Enter your message"): | |
| umsg = Conversation(message) | |
| answer = convo(umsg) | |
| with st.chat_message("assistant"): | |
| st.write(answer) | |
| chatbot() |