seonglae commited on
Commit
f12e339
1 Parent(s): 9887b4f

fix: remove question mark for better inference

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -9,7 +9,7 @@ from model import summarize_text, get_summarizer
9
  from model import ask_reader, get_reader
10
 
11
  TITLE = 'ReSRer: Retriever-Summarizer-Reader'
12
- INITIAL = "What is the population of NYC?"
13
 
14
  st.set_page_config(page_title=TITLE)
15
  st.header(TITLE)
@@ -57,12 +57,12 @@ st.markdown(styl, unsafe_allow_html=True)
57
 
58
  question = st.text_input("Question", INITIAL)
59
  col1, col2, col3 = st.columns(3)
60
- if col1.button("What is the capital of South Korea?"):
61
- question = "What is the capital of South Korea?"
62
- if col2.button("What is the most famous building in Paris?"):
63
- question = "What is the most famous building in Paris?"
64
- if col3.button("Who is the actor of Harry Potter?"):
65
- question = "Who is the actor of Harry Potter?"
66
 
67
 
68
  @torch.inference_mode()
 
9
  from model import ask_reader, get_reader
10
 
11
  TITLE = 'ReSRer: Retriever-Summarizer-Reader'
12
+ INITIAL = "What is the population of NYC"
13
 
14
  st.set_page_config(page_title=TITLE)
15
  st.header(TITLE)
 
57
 
58
  question = st.text_input("Question", INITIAL)
59
  col1, col2, col3 = st.columns(3)
60
+ if col1.button("What is the capital of South Korea"):
61
+ question = "What is the capital of South Korea"
62
+ if col2.button("What is the most famous building in Paris"):
63
+ question = "What is the most famous building in Paris"
64
+ if col3.button("Who is the actor of Harry Potter"):
65
+ question = "Who is the actor of Harry Potter"
66
 
67
 
68
  @torch.inference_mode()