Spaces:
Sleeping
Sleeping
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()
|