Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -13,10 +13,15 @@ from get_embedding_function import get_embedding_function
|
|
13 |
from langchain.document_loaders.pdf import PyPDFDirectoryLoader
|
14 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
15 |
from langchain.schema.document import Document
|
|
|
|
|
|
|
|
|
|
|
16 |
|
17 |
app = Flask(__name__)
|
18 |
UPLOAD_FOLDER = '/code/uploads'
|
19 |
-
CHROMA_PATH =
|
20 |
if not os.path.exists(UPLOAD_FOLDER):
|
21 |
os.makedirs(UPLOAD_FOLDER)
|
22 |
|
|
|
13 |
from langchain.document_loaders.pdf import PyPDFDirectoryLoader
|
14 |
from langchain_text_splitters import RecursiveCharacterTextSplitter
|
15 |
from langchain.schema.document import Document
|
16 |
+
import tempfile
|
17 |
+
|
18 |
+
# Create a temporary directory for Chroma if running in Hugging Face Spaces
|
19 |
+
|
20 |
+
|
21 |
|
22 |
app = Flask(__name__)
|
23 |
UPLOAD_FOLDER = '/code/uploads'
|
24 |
+
CHROMA_PATH = tempfile.mkdtemp() # Use the same folder for Chroma
|
25 |
if not os.path.exists(UPLOAD_FOLDER):
|
26 |
os.makedirs(UPLOAD_FOLDER)
|
27 |
|