Update app.py

app.py

@@ -4,27 +4,26 @@ from transformers import TFGPT2LMHeadModel ,GPT2Tokenizer, BitsAndBytesConfig
 
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
 model = TFGPT2LMHeadModel.from_pretrained('gpt2',pad_token_id = tokenizer.eos_token_id)
-
 def generate(inp):
     input_ids = tokenizer.encode(inp,return_tensors = 'tf')
     beam_output =  model.generate(input_ids, max_length = 90,num_beams = 5, no_repeat_ngram_size = 2, early_stopping = True)
     output = tokenizer.decode(beam_output[0],skip_special_tokens = True, clean_up_tokenization_spaces = True)
     return ".".join(output.split(".")[:-1]) + "."
     
-st.title("Programming Bot")
+st.title("Animal Bot")
 if "messages" not in st.session_state:
     st.session_state.messages = []
     st.session_state.messages.append({
         'role':'assistant',
-        'content':"Hi! I'm your Programming Bot assistant"
+        'content':"Hi! I'm your Animal assistant, any queries about animals ?"
     })
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 prompt = st.chat_input("Any Queries?")
 if prompt:
-    # with st.chat_message("user"):
-        # st.markdown(prompt)
+    with st.chat_message("user"):
+        st.markdown(prompt)
     st.session_state.messages.append({"role":"user","content":prompt})
     response = generate(prompt)
     with st.chat_message("assistant"):