File size: 973 Bytes
29817df 208053e a622bac 29817df a1cfd13 29817df 5074b67 29817df 5fbc8cc 29817df a1cfd13 5fbc8cc 29817df 5fbc8cc c2eb4e6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 | import streamlit as st
from transformers import BlenderbotTokenizer, BlenderbotForConditionalGeneration
model_str = 'facebook/blenderbot-400M-distill'
tokenizer = BlenderbotTokenizer.from_pretrained(model_str)
model = BlenderbotForConditionalGeneration.from_pretrained(model_str)
def chatbot(text):
inputs = tokenizer([text], return_tensors='pt')
response = model.generate(
inputs['input_ids'],
attention_mask=inputs['attention_mask'],
max_length=100
)
output = tokenizer.decode(response[0], skip_special_tokens=True)
return output
def main():
st.title("convo-bot")
user_input = st.text_input("user:")
if user_input:
bot_response = chatbot(user_input)
st.text_area("bot:", value=bot_response, height=200)
if __name__ == '__main__':
main()
# https://huggingface.co/docs/transformers/model_doc/blenderbot
# https://huggingface.co/facebook/blenderbot-400M-distill
# https://streamlit.io/ |