Spaces:
Paused
Paused
Daniel Marques
commited on
Commit
·
6318d9a
1
Parent(s):
d5863e7
feat: add message tamplate
Browse files
main.py
CHANGED
@@ -23,7 +23,6 @@ from prompt_template_utils import get_prompt_template
|
|
23 |
|
24 |
# from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
|
25 |
from langchain.vectorstores import Chroma
|
26 |
-
from werkzeug.utils import secure_filename
|
27 |
|
28 |
from constants import CHROMA_SETTINGS, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY, MODEL_ID, MODEL_BASENAME
|
29 |
|
@@ -164,7 +163,7 @@ async def create_upload_file(file: Union[UploadFile, None] = None):
|
|
164 |
if file.filename == "":
|
165 |
raise HTTPException(status_code=400, detail="No selected file")
|
166 |
if file:
|
167 |
-
filename =
|
168 |
folder_path = "SOURCE_DOCUMENTS"
|
169 |
|
170 |
if not os.path.exists(folder_path):
|
|
|
23 |
|
24 |
# from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
|
25 |
from langchain.vectorstores import Chroma
|
|
|
26 |
|
27 |
from constants import CHROMA_SETTINGS, EMBEDDING_MODEL_NAME, PERSIST_DIRECTORY, MODEL_ID, MODEL_BASENAME
|
28 |
|
|
|
163 |
if file.filename == "":
|
164 |
raise HTTPException(status_code=400, detail="No selected file")
|
165 |
if file:
|
166 |
+
filename = file.filename
|
167 |
folder_path = "SOURCE_DOCUMENTS"
|
168 |
|
169 |
if not os.path.exists(folder_path):
|