Update pages/answers.py

pages/answers.py

@@ -1,7 +1,15 @@
 import streamlit as st
 from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
+from streamlit_extras.let_it_rain import rain
 
-model_name = "timpal0l/mdeberta-v3-base-squad2"
+rain(
+    emoji="❔",
+    font_size=54,
+    falling_speed=5,
+    animation_length="infinite",
+)
+
+model_name = "deepset/roberta-base-squad2"
 model = AutoModelForQuestionAnswering.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
@@ -13,15 +21,16 @@ def get_answer(context, question):
     return answer
 
 def main():
-    st.title("Question Answering App")
-    st.markdown("Enter the context and question, then click on 'Get Answer' to retrieve the answer.")
+    st.title("Question Answering App :robot_face:")
+    st.divider()
+    st.markdown("### **Enter the context and question, then click on ':blue[Get Answer]' to retrieve the answer:**")
 
     
-    context = st.text_area("Context", "Enter the context here...")
-    question = st.text_input("Question", "Enter the question here...")
+    context = st.text_area("**:blue[Context]**", "Enter the context here...")
+    question = st.text_input("**:blue[Question]**", "Enter the question here...")
 
     
-    if st.button("Get Answer"):
+    if st.button(":blue[**Get Answer**]"):
         
         if context.strip() == "" or question.strip() == "":
             st.warning("Please enter the context and question.")