Spaces:
Sleeping
Sleeping
add papers dropdown
Browse files- 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.
|
| 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=
|
| 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:
|