Raghav001 commited on
Commit
df1cbf3
1 Parent(s): 1b4656d
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,11 +4,12 @@ import gradio as gr
4
  # from concurrent.futures import ThreadPoolExecutor
5
  import pdfplumber
6
  import pandas as pd
 
7
  import time
8
  from cnocr import CnOcr
9
 
10
  # from langchain.document_loaders import PyPDFLoader
11
-
12
  from sentence_transformers import SentenceTransformer, models, util
13
  word_embedding_model = models.Transformer('sentence-transformers/all-MiniLM-L6-v2', do_lower_case=True)
14
  pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(), pooling_mode='cls')
 
4
  # from concurrent.futures import ThreadPoolExecutor
5
  import pdfplumber
6
  import pandas as pd
7
+ import langchain
8
  import time
9
  from cnocr import CnOcr
10
 
11
  # from langchain.document_loaders import PyPDFLoader
12
+ from langchain.document_loaders import UnstructuredWordDocumentLoader
13
  from sentence_transformers import SentenceTransformer, models, util
14
  word_embedding_model = models.Transformer('sentence-transformers/all-MiniLM-L6-v2', do_lower_case=True)
15
  pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(), pooling_mode='cls')