farmax commited on
Commit
385cfb6
·
verified ·
1 Parent(s): f0ea8fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from googletrans import Translator
4
  import requests
5
  from dotenv import load_dotenv
6
  import numpy as np
7
- from langchain.embeddings import Chroma
8
  from langchain.vectorstores import Chroma
9
  from langchain.document_loaders import UnstructuredPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
@@ -20,6 +20,9 @@ from tqdm import tqdm
20
  import torch
21
  import logging
22
 
 
 
 
23
  logging.basicConfig(level=logging.INFO)
24
  logger = logging.getLogger(__name__)
25
 
 
4
  import requests
5
  from dotenv import load_dotenv
6
  import numpy as np
7
+ from langchain.embeddings import HuggingFaceEmbeddings
8
  from langchain.vectorstores import Chroma
9
  from langchain.document_loaders import UnstructuredPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
 
20
  import torch
21
  import logging
22
 
23
+ # Replace the old Chroma import with HuggingFaceEmbeddings
24
+ embedding_function = HuggingFaceEmbeddings()
25
+
26
  logging.basicConfig(level=logging.INFO)
27
  logger = logging.getLogger(__name__)
28