Update app.py

app.py

@@ -45,7 +45,8 @@ class DocumentSearch:
         return sorted(results, key=lambda x: x['score'], reverse=True)[:k]
 
 
-if __name__ == "__main__":
+@st.cache(suppress_st_warning=True)
+def get_surfer():
     enc_path = "ivan-savchuk/msmarco-distilbert-dot-v5-tuned-full-v1"
     idx_path = "idx_vectors.index"
     cross_enc_path = "ivan-savchuk/cross-encoder-ms-marco-MiniLM-L-12-v2-tuned_mediqa-v1"
@@ -57,6 +58,12 @@ if __name__ == "__main__":
         index_path=idx_path,
         cross_encoder_path=cross_enc_path
     )
+    return surfer
+
+
+if __name__ == "__main__":
+    # get instance of DocumentSearch class
+    surfer = get_surfer()
     # streamlit part starts here with title
     st.title('Medical Search')
     # input form