shripadbhat commited on
Commit
1108b82
·
1 Parent(s): 5e8df6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -45,7 +45,7 @@ def fetch_answers(question, document ):
45
  return top_5_query_paragraph_answer_list
46
 
47
  st.title('Document Question Answering System')
48
- st.text("Loading the models...")
49
  my_bar = st.progress(0)
50
  model_name = "MaRiOrOsSi/t5-base-finetuned-question-answering"
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -57,9 +57,9 @@ passage_retreival_model = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
57
  my_bar.progress(75)
58
  qa_model = pipeline("question-answering",'a-ware/bart-squadv2')
59
  my_bar.progress(100)
60
- st.text('Models Loaded')
61
 
62
- query = st.text_area("Query", "", height=25)
63
  document = st.text_area("Document Text", "", height=100)
64
 
65
  if st.button("Get Answers From Document"):
 
45
  return top_5_query_paragraph_answer_list
46
 
47
  st.title('Document Question Answering System')
48
+ st.write("Loading the models...")
49
  my_bar = st.progress(0)
50
  model_name = "MaRiOrOsSi/t5-base-finetuned-question-answering"
51
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
57
  my_bar.progress(75)
58
  qa_model = pipeline("question-answering",'a-ware/bart-squadv2')
59
  my_bar.progress(100)
60
+ st.write('Models Loaded')
61
 
62
+ query = st.text_input("Query")
63
  document = st.text_area("Document Text", "", height=100)
64
 
65
  if st.button("Get Answers From Document"):