Update app.py

app.py

@@ -15,17 +15,23 @@ class DocumentSearch:
         sbert: encoder
         sbert: cross_encoder
     '''
-    def __init__(self, labels_path: str, encoder_path: str,
-                 index_path: str, cross_encoder_path: str):
+    # we mention pass to every file that needed to run models
+    # and search over our data
+    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"
+    docs_path = "docs.json"
+    
+    def __init__(self):
         # loading docs and corresponding urls
         with open(labels_path, 'r') as json_file:
-            self.docs = json.load(json_file)
+            self.docs = json.load(DocumentSearch.docs_path)
         # loading sbert encoder model
-        self.encoder = SentenceTransformer(encoder_path)
+        self.encoder = SentenceTransformer(DocumentSearch.enc_path)
         # loading faiss index
-        self.index = faiss.read_index(index_path)
+        self.index = faiss.read_index(DocumentSearch.idx_path)
         # loading sbert cross_encoder
-        self.cross_encoder = CrossEncoder(cross_encoder_path)
+        self.cross_encoder = CrossEncoder(DocumentSearch.cross_enc_path)
 
     def search(self, query: str, k: int) -> list:
         # get vector representation of text query
@@ -45,25 +51,9 @@ class DocumentSearch:
         return sorted(results, key=lambda x: x['score'], reverse=True)[:k]
 
 
-@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"
-    docs_path = "docs.json"
-    # get instance of DocumentSearch class
-    surfer = DocumentSearch(
-        labels_path=docs_path,
-        encoder_path=enc_path,
-        index_path=idx_path,
-        cross_encoder_path=cross_enc_path
-    )
-    return surfer
-
-
 if __name__ == "__main__":
     # get instance of DocumentSearch class
-    surfer = get_surfer()
+    surfer = DocumentSearch()
     # streamlit part starts here with title
     st.title('Medical Search')
     # input form