Update app.py

app.py

@@ -17,6 +17,7 @@ class SystemMessage:
 model_name = "facebook/bart-base"  # Replace with your choice (e.g.,t5-small or facebook/bart-base or EleutherAI/gpt-neo-125M)
 
 # Load the model and tokenizer
+
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
@@ -40,14 +41,12 @@ class AIMessage:
 
 
 def load_answer(question):
-
-  st.session_state.sessionMessages.append(HumanMessage(content=question))
-
-  inputs = tokenizer(question, return_tensors="pt")
-  outputs = model.generate(**inputs)
-  response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
-  st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
-  return assistant_answer.content
+    st.session_state.sessionMessages.append(HumanMessage(content=question))
+    inputs = tokenizer(question, return_tensors="pt")  # Tokenize the question
+    outputs = model.generate(**inputs)  # Generate response based on tokenized input
+    response_text = tokenizer.decode(outputs[0], skip_special_tokens=True)  # Decode the output
+    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
+    return assistant_answer.content
 
 
 
@@ -59,7 +58,6 @@ def get_text():
 
 
 
-
 user_input=get_text()
 submit = st.button('Generate')