Update app.py

app.py

@@ -130,7 +130,7 @@ def bi_encode(bi_enc,passages):
 
     st.success(f"Embeddings computed. Shape: {corpus_embeddings.shape}")
     
-    return corpus_embeddings
+    return bi_encoder, corpus_embeddings
     
 @st.cache(allow_output_mutation=True)
 def cross_encode():
@@ -317,7 +317,7 @@ if search:
         with st.spinner(
             text=f"Loading {bi_encoder_type} bi-encoder and embedding document into vector space. This might take a few seconds depending on the length of your document..."
         ):
-            corpus_embeddings = bi_encode(bi_encoder_type,passages)
+            bi_encoder, corpus_embeddings = bi_encode(bi_encoder_type,passages)
             cross_encoder = cross_encode()
             bm25 = bm25_api(passages)