Spaces:
Runtime error
Runtime error
Sheshera Mysore
commited on
Commit
·
0cdb40c
1
Parent(s):
ab5080f
Add a spinner for generating the profile and recs.
Browse files
app.py
CHANGED
|
@@ -416,11 +416,13 @@ if uploaded_file is not None:
|
|
| 416 |
kp2val_vectors = st.session_state.kp2val_vectors
|
| 417 |
user_tplan = st.session_state.user_tplan
|
| 418 |
else: # This happens on the first run.
|
| 419 |
-
|
|
|
|
| 420 |
st.session_state['kp2val_vectors'] = kp2val_vectors
|
| 421 |
st.session_state['user_tplan'] = user_tplan
|
| 422 |
else:
|
| 423 |
-
|
|
|
|
| 424 |
st.session_state['kp2val_vectors'] = kp2val_vectors
|
| 425 |
st.session_state['user_tplan'] = user_tplan
|
| 426 |
st.session_state.run_user_kps.append(copy.copy(input_user_kps))
|
|
@@ -449,11 +451,12 @@ if uploaded_file is not None:
|
|
| 449 |
# st.write('Generating recs...')
|
| 450 |
st.session_state.tuning_i += 1
|
| 451 |
st.session_state.i_selections.append(copy.deepcopy(profile_selections))
|
| 452 |
-
|
| 453 |
-
|
| 454 |
-
|
| 455 |
-
|
| 456 |
-
|
|
|
|
| 457 |
st.session_state.i_resultps[st.session_state.tuning_i] = copy.deepcopy(top_papers)
|
| 458 |
|
| 459 |
# Read off from the result cache and allow users to save some papers.
|
|
|
|
| 416 |
kp2val_vectors = st.session_state.kp2val_vectors
|
| 417 |
user_tplan = st.session_state.user_tplan
|
| 418 |
else: # This happens on the first run.
|
| 419 |
+
with st.spinner(text="Generating profile..."):
|
| 420 |
+
kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
|
| 421 |
st.session_state['kp2val_vectors'] = kp2val_vectors
|
| 422 |
st.session_state['user_tplan'] = user_tplan
|
| 423 |
else:
|
| 424 |
+
with st.spinner(text="Generating profile..."):
|
| 425 |
+
kp2val_vectors, user_tplan = generate_profile_values(profile_keyphrases=input_user_kps)
|
| 426 |
st.session_state['kp2val_vectors'] = kp2val_vectors
|
| 427 |
st.session_state['user_tplan'] = user_tplan
|
| 428 |
st.session_state.run_user_kps.append(copy.copy(input_user_kps))
|
|
|
|
| 451 |
# st.write('Generating recs...')
|
| 452 |
st.session_state.tuning_i += 1
|
| 453 |
st.session_state.i_selections.append(copy.deepcopy(profile_selections))
|
| 454 |
+
with st.spinner(text="Recommending papers..."):
|
| 455 |
+
top_papers = second_stage_ranked_docs(first_stage_pids=first_stage_ret_pids,
|
| 456 |
+
selected_query_kps=profile_selections,
|
| 457 |
+
pid2abstract=pid2abstract_cands,
|
| 458 |
+
pid2sent_reps_cand=pid2sent_vectors_cands,
|
| 459 |
+
to_rank=30)
|
| 460 |
st.session_state.i_resultps[st.session_state.tuning_i] = copy.deepcopy(top_papers)
|
| 461 |
|
| 462 |
# Read off from the result cache and allow users to save some papers.
|