Added scikit learn to requirements

app.py

@@ -6,7 +6,6 @@ import tensorflow_hub as hub
 import openai
 import gradio as gr
 import os
-from tqdm.auto import tqdm
 from sklearn.neighbors import NearestNeighbors
 
 
@@ -29,7 +28,7 @@ def pdf_to_text(path, start_page=1, end_page=None):
 
     text_list = []
 
-    for i in tqdm(range(start_page-1, end_page)):
+    for i in range(start_page-1, end_page):
         text = doc.load_page(i).get_text("text")
         text = preprocess(text)
         text_list.append(text)
@@ -84,7 +83,7 @@ class SemanticSearch:
     
     def get_text_embedding(self, texts, batch=1000):
         embeddings = []
-        for i in tqdm(range(0, len(texts), batch)):
+        for i in range(0, len(texts), batch):
             text_batch = texts[i:(i+batch)]
             emb_batch = self.use(text_batch)
             embeddings.append(emb_batch)

requirements.txt

@@ -1,4 +1,5 @@
 PyMuPDF
 openai
 tensorflow==2.9.2
-tensorflow-hub==0.12.0
+tensorflow-hub==0.12.0
+scikit-learn==1.0.2