File size: 388 Bytes
3a66a27 43dad90 d2d374e 3a66a27 6ecd566 43dad90 d2d374e 3eccd00 43dad90 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
import streamlit as st
import os
from transformers import pipeline, Conversation
convo = pipeline(model="facebook/blenderbot-400M-distill")
def chatbot():
st.title("ChatBot")
if message := st.chat_input("Enter your query"):
umsg = Conversation(message)
answer = convo(umsg)
with st.chat_message("Assistant"):
st.write(answer)
chatbot()
|