Spaces:
Running
Running
minor fixes
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
|
@@ -150,7 +150,7 @@ def load_papers_from_rds():
|
|
| 150 |
return []
|
| 151 |
|
| 152 |
|
| 153 |
-
#
|
| 154 |
def search_theorems(query, model, theorems_data, embeddings_db):
|
| 155 |
"""
|
| 156 |
Takes a user query and finds the top 10 most similar theorems.
|
|
@@ -163,7 +163,7 @@ def search_theorems(query, model, theorems_data, embeddings_db):
|
|
| 163 |
cosine_scores = util.cos_sim(query_embedding, embeddings_db)[0]
|
| 164 |
top_results_indices = np.argsort(-cosine_scores.cpu())[:10]
|
| 165 |
|
| 166 |
-
st.subheader("Top
|
| 167 |
|
| 168 |
if len(top_results_indices) == 0:
|
| 169 |
st.write("No results found.")
|
|
|
|
| 150 |
return []
|
| 151 |
|
| 152 |
|
| 153 |
+
# The Search Function
|
| 154 |
def search_theorems(query, model, theorems_data, embeddings_db):
|
| 155 |
"""
|
| 156 |
Takes a user query and finds the top 10 most similar theorems.
|
|
|
|
| 163 |
cosine_scores = util.cos_sim(query_embedding, embeddings_db)[0]
|
| 164 |
top_results_indices = np.argsort(-cosine_scores.cpu())[:10]
|
| 165 |
|
| 166 |
+
st.subheader("Top 10 Most Similar Theorems")
|
| 167 |
|
| 168 |
if len(top_results_indices) == 0:
|
| 169 |
st.write("No results found.")
|