# -*- coding: utf-8 -*- """ Created on Thu Sep 21 22:17:43 2023 @author: Loges """ import streamlit as st import sentencepiece from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/results_T5").to("cpu") tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/results_T5") pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer) st.set_page_config(page_title='Sample Chatbot', layout='wide') if 'messages' not in st.session_state: st.session_state.messages=[] st.subheader("Mine AI") for message in st.session_state.messages: with st.chat_message(message['role']): st.markdown(message['content']) ## messages element format: {'role':'user', 'content':''} if prompt:=st.chat_input("What is up!"): with st.chat_message("user"): st.markdown(prompt) st.session_state.messages.append({ 'role':'user', 'content': prompt}) out=pipe(prompt) response = f"Analysis: {out}" with st.chat_message("assistant"): st.markdown(response) st.session_state.messages.append({ 'role':'assistant', 'content': response})