Update app.py

app.py

@@ -99,7 +99,7 @@ def preprocess_plain_text(text, window_size=3):
         if len(paragraph.strip()) > 0:
             paragraphs.append(sent_tokenize(paragraph.strip()))
 
-    window_size = window_size
+    window_size = 3
     passages = []
     for paragraph in paragraphs:
         for start_idx in range(0, len(paragraph), window_size):
@@ -152,6 +152,7 @@ def display_as_table(model, top_k=2, score='score'):
 # Streamlit App
 
 st.title("Search with Retrieve & Rerank")
+window_size = 3
 
 
 # This function will search all wikipedia articles for passages that answer the query
@@ -225,12 +226,12 @@ search_query = st.text_input("Please Enter your search query here",
 if validators.url(url_text):
     # if input is URL
     title, text = extract_text_from_url(url_text)
-    passages = preprocess_plain_text(text, window_size=window_size)
+    passages = preprocess_plain_text(text, window_size=3)
 
 elif upload_doc:
 
     text, pdf_title = extract_text_from_file(upload_doc)
-    passages = preprocess_plain_text(text, window_size=window_size)
+    passages = preprocess_plain_text(text, window_size=3)
 
 col1, col2 = st.columns(2)