Update app.py

app.py

@@ -4,13 +4,13 @@ Created on Sun Oct 01 10:49:43 2023
 """
 
 import streamlit as st
-# import sentencepiece
-# from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
+import sentencepiece
+from transformers import pipeline, T5Tokenizer, T5ForConditionalGeneration
 
-# model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/AI_Chaperone").to("cpu")
-# tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/AI_Chaperone")
+model=T5ForConditionalGeneration.from_pretrained("Logeswaransr/AI_Chaperone").to("cpu")
+tokenizer=T5Tokenizer.from_pretrained("Logeswaransr/AI_Chaperone")
 
-# pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
+pipe=pipeline('text2text-generation', model=model, tokenizer=tokenizer)
 
 greetings=["Hello!","I am AI Chaperone. Your special virtual assistant for your needs.", "Feel free to ask me anything. I will do what I can."]
 
@@ -39,14 +39,14 @@ if prompt:=st.chat_input("Enter your query"):
         'role':'user',
         'content': prompt})
 
-    # out=pipe(prompt)
-    # response=out[0]['generated_text']
+    out=pipe(prompt)
+    response=out[0]['generated_text']
     
     # response = f"Analysis: {response}"
     
     with st.chat_message("assistant"):
-        st.markdown(prompt)
+        st.markdown(response)
         
     st.session_state.messages.append({
         'role':'assistant',
-        'content': prompt})
+        'content': response})