Update app.py

app.py

@@ -56,6 +56,11 @@ chunk_size = 2048
 
 print("Finish the model init process")
 
+def get_text_embedding(text):
+
+    return embeddings.embed_query(text)
+
+
 def extract_text(file):
 
     
@@ -75,8 +80,10 @@ def extract_text(file):
 def summarise(text):
 
     chunks = [text[i:i + chunk_size] for i in range(0, len(text), chunk_size)]
+
+    text_embeddings = np.array([get_text_embedding(chunk) for chunk in chunks])
     
-    return len(chunks)
+    return len(text_embeddings)
 
 with gr.Blocks() as demo:
     file_input = gr.File(label="Upload a PDF file")