livctr commited on
Commit
faa63d0
·
1 Parent(s): bbe4dd0

add papers dropdown

Browse files
Files changed (1) hide show
  1. USMLPhDRecommender.py +8 -3
USMLPhDRecommender.py CHANGED
@@ -4,8 +4,7 @@ from core.recommender import EmbeddingProcessor, Recommender
4
 
5
  st.title("U.S. ML PhD Faculty Advisor Recommender")
6
 
7
- st.write("See the code on [GitHub](https://github.com/livctr/us-ml-phd-rec.git) on a disclaimer and how to use.")
8
- st.write("")
9
 
10
  # Set up
11
  embedding_processor = EmbeddingProcessor()
@@ -14,9 +13,15 @@ recommender = Recommender(embedding_processor)
14
  # Query input field
15
  query = st.text_input("Name an ML research area you are interested in (e.g. low-rank adaptation)")
16
 
 
 
 
 
 
 
17
  # Search and display professors
18
  if query:
19
- top_k_indices = recommender.get_top_k(query, top_k=10)
20
  professors_data = recommender.get_recommended_data(top_k_indices)
21
 
22
  if professors_data:
 
4
 
5
  st.title("U.S. ML PhD Faculty Advisor Recommender")
6
 
7
+ st.markdown("See the [GitHub](https://github.com/livctr/us-ml-phd-rec.git) for an **important disclaimer** and how to use.")
 
8
 
9
  # Set up
10
  embedding_processor = EmbeddingProcessor()
 
13
  # Query input field
14
  query = st.text_input("Name an ML research area you are interested in (e.g. low-rank adaptation)")
15
 
16
+ num_papers = st.selectbox(
17
+ "Select the number of papers to display",
18
+ options=[5, 10, 20, 50, 100],
19
+ index=1 # default value set to 10
20
+ )
21
+
22
  # Search and display professors
23
  if query:
24
+ top_k_indices = recommender.get_top_k(query, top_k=num_papers)
25
  professors_data = recommender.get_recommended_data(top_k_indices)
26
 
27
  if professors_data: