Neurolingua commited on
Commit
573cef7
1 Parent(s): d7c3478

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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 = UPLOAD_FOLDER # Use the same folder for Chroma
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