Update app.py

app.py

@@ -13,7 +13,7 @@ st_model_load = st.text('Loading wikipedia embedding...')
 def load_embedding():
     print("Loading embedding...")
     hf_hub_download(repo_id="abokbot/wikipedia-embedding", filename="simple_wikipedia_embedding.pt")
-    wikipedia_embedding = torch.load("wikipedia-embedding/simple_wikipedia_embedding.pt") 
+    wikipedia_embedding = torch.load("simple_wikipedia_embedding.pt") 
     print("Embedding loaded!")
     return wikipedia_embedding
 
@@ -21,6 +21,25 @@ wikipedia_embedding = load_embedding()
 st.success('Embedding loaded!')
 st_model_load.text("")
 
+def load_encoders():
+    print("Loading encoders...")
+    bi_encoder = SentenceTransformer('msmarco-MiniLM-L-6-v3')
+    bi_encoder.max_seq_length = 256     #Truncate long passages to 256 tokens
+    top_k = 32  
+    cross_encoder = CrossEncoder('cross-encoder/ms-marco-TinyBERT-L-2-v2')
+    return bi_encoder, cross_encoder
+
+bi_encoder, cross_encoder = load_encoders()
+st.success('Encoders loaded!')
+st_model_load.text("")
+    
+if 'text' not in st.session_state:
+    st.session_state.text = ""
+st_text_area = st.text_area(
+    'Enter query (e.g. What is the capital city of Kenya? or Number of deputees in French parliement)', 
+    value=st.session_state.text, 
+    height=100
+)
 """