update

app.py

@@ -41,7 +41,6 @@ def load_title_embeddings():
     return title_embeddings
 
 
-@st.cache_data
 def get_retrieval_results(index, input_text, top_k, tokenizer, title_df):
     batch_dict = tokenizer(f"query: {input_text}", max_length=512, padding=True, truncation=True, return_tensors='pt')
     with torch.no_grad():
@@ -70,8 +69,8 @@ if __name__ == "__main__":
     index = faiss.IndexFlatL2(1024)
     index.add(title_embeddings)
     
-    st.markdown("## NLP2024 類似論文検索")
-    input_text = st.text_input('input', '', placeholder='ここに論文のタイトルを入力してください')
+    st.markdown("## NLP2024 論文検索")
+    input_text = st.text_input('query', '', placeholder='')
     top_k = st.number_input('top_k', min_value=1, value=10, step=1)
     
     if st.button('検索'):