vasevooo commited on
Commit
df6367f
·
1 Parent(s): d02a704

Update pages/answers.py

Browse files
Files changed (1) hide show
  1. pages/answers.py +15 -6
pages/answers.py CHANGED
@@ -1,7 +1,15 @@
1
  import streamlit as st
2
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
 
3
 
4
- model_name = "timpal0l/mdeberta-v3-base-squad2"
 
 
 
 
 
 
 
5
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
7
 
@@ -13,15 +21,16 @@ def get_answer(context, question):
13
  return answer
14
 
15
  def main():
16
- st.title("Question Answering App")
17
- st.markdown("Enter the context and question, then click on 'Get Answer' to retrieve the answer.")
 
18
 
19
 
20
- context = st.text_area("Context", "Enter the context here...")
21
- question = st.text_input("Question", "Enter the question here...")
22
 
23
 
24
- if st.button("Get Answer"):
25
 
26
  if context.strip() == "" or question.strip() == "":
27
  st.warning("Please enter the context and question.")
 
1
  import streamlit as st
2
  from transformers import AutoModelForQuestionAnswering, AutoTokenizer, pipeline
3
+ from streamlit_extras.let_it_rain import rain
4
 
5
+ rain(
6
+ emoji="❔",
7
+ font_size=54,
8
+ falling_speed=5,
9
+ animation_length="infinite",
10
+ )
11
+
12
+ model_name = "deepset/roberta-base-squad2"
13
  model = AutoModelForQuestionAnswering.from_pretrained(model_name)
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
15
 
 
21
  return answer
22
 
23
  def main():
24
+ st.title("Question Answering App :robot_face:")
25
+ st.divider()
26
+ st.markdown("### **Enter the context and question, then click on ':blue[Get Answer]' to retrieve the answer:**")
27
 
28
 
29
+ context = st.text_area("**:blue[Context]**", "Enter the context here...")
30
+ question = st.text_input("**:blue[Question]**", "Enter the question here...")
31
 
32
 
33
+ if st.button(":blue[**Get Answer**]"):
34
 
35
  if context.strip() == "" or question.strip() == "":
36
  st.warning("Please enter the context and question.")