Update Rock_fact_checker.py
Browse files- Rock_fact_checker.py +5 -5
Rock_fact_checker.py
CHANGED
|
@@ -22,17 +22,17 @@ def main():
|
|
| 22 |
build_sidebar()
|
| 23 |
|
| 24 |
# Persistent state
|
| 25 |
-
set_state_if_absent("statement", "
|
| 26 |
set_state_if_absent("answer", "")
|
| 27 |
set_state_if_absent("results", None)
|
| 28 |
set_state_if_absent("raw_json", None)
|
| 29 |
set_state_if_absent("random_statement_requested", False)
|
| 30 |
|
| 31 |
-
st.write("
|
| 32 |
st.write()
|
| 33 |
st.markdown(
|
| 34 |
"""
|
| 35 |
-
##### Enter
|
| 36 |
"""
|
| 37 |
)
|
| 38 |
# Search bar
|
|
@@ -78,7 +78,7 @@ def main():
|
|
| 78 |
time_start = time.time()
|
| 79 |
reset_results()
|
| 80 |
st.session_state.statement = statement
|
| 81 |
-
with st.spinner("🧠
|
| 82 |
try:
|
| 83 |
st.session_state.results = query(statement, RETRIEVER_TOP_K)
|
| 84 |
print(f"S: {statement}")
|
|
@@ -116,7 +116,7 @@ def main():
|
|
| 116 |
st.markdown(f"###### Most Relevant snippets:")
|
| 117 |
df, urls = create_df_for_relevant_snippets(docs)
|
| 118 |
st.dataframe(df)
|
| 119 |
-
str_wiki_pages = "
|
| 120 |
for doc, url in urls.items():
|
| 121 |
str_wiki_pages += f"[{doc}]({url}) "
|
| 122 |
st.markdown(str_wiki_pages)
|
|
|
|
| 22 |
build_sidebar()
|
| 23 |
|
| 24 |
# Persistent state
|
| 25 |
+
set_state_if_absent("statement", "Referral bonus can only be given if your friend joins Newton School on your behalf")
|
| 26 |
set_state_if_absent("answer", "")
|
| 27 |
set_state_if_absent("results", None)
|
| 28 |
set_state_if_absent("raw_json", None)
|
| 29 |
set_state_if_absent("random_statement_requested", False)
|
| 30 |
|
| 31 |
+
st.write("Referral Mis-Sell")
|
| 32 |
st.write()
|
| 33 |
st.markdown(
|
| 34 |
"""
|
| 35 |
+
##### Enter statement
|
| 36 |
"""
|
| 37 |
)
|
| 38 |
# Search bar
|
|
|
|
| 78 |
time_start = time.time()
|
| 79 |
reset_results()
|
| 80 |
st.session_state.statement = statement
|
| 81 |
+
with st.spinner("🧠 Running Model..."):
|
| 82 |
try:
|
| 83 |
st.session_state.results = query(statement, RETRIEVER_TOP_K)
|
| 84 |
print(f"S: {statement}")
|
|
|
|
| 116 |
st.markdown(f"###### Most Relevant snippets:")
|
| 117 |
df, urls = create_df_for_relevant_snippets(docs)
|
| 118 |
st.dataframe(df)
|
| 119 |
+
str_wiki_pages = "Data: "
|
| 120 |
for doc, url in urls.items():
|
| 121 |
str_wiki_pages += f"[{doc}]({url}) "
|
| 122 |
st.markdown(str_wiki_pages)
|