Spaces:
Running
Running
fix: remove question mark for better inference
Browse files
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()
|