Update app.py

app.py

@@ -81,8 +81,7 @@ def search():
     print("Top-3 Cross-Encoder Re-ranker hits")
     results = []
     for hit in hits[:3]:
-        results.append
-        (
+        results.append(
             {
                 "score": round(hit['cross-score'], 3),
                 "title": dataset[hit['corpus_id']]["title"],
@@ -90,17 +89,14 @@ def search():
                 "link": dataset[hit['corpus_id']]["url"]
             }
         )
-    st.session_state.results = results
+    return results
 
 
 # search button
-st_search_button = st.button('Search', on_click=search)
+st_search_button = st.button('Search')
+if st_search_button:
+    results = search()
 
-if 'results' not in st.session_state:
-    st.session_state.results = {}
-
-if len(st.session_state.results) > 0:
-    with st.container():
-        st.subheader("Search results")
-        for result in st.session_state.results:
-            st.text(result)
+st.subheader("Search results")
+for result in results:
+    st.text(result)