slszeto commited on
Commit
f83c831
·
verified ·
1 Parent(s): 6de8c39

minor fixes

Browse files
Files changed (1) hide show
  1. 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
- # --- 3. 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,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 5 Most Similar Theorems")
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.")