danicafisher commited on
Commit
e5beda5
1 Parent(s): be3813f

Update helper_functions.py

Browse files
Files changed (1) hide show
  1. helper_functions.py +2 -2
helper_functions.py CHANGED
@@ -4,7 +4,7 @@ import os
4
 
5
  def process_file(uploaded_file):
6
  # save the file temporarily
7
- temp_file = "./temp.pdf"
8
  with open(temp_file, "wb") as file:
9
  file.write(uploaded_file.content)
10
  file_name = uploaded_file.name
@@ -15,7 +15,7 @@ def process_file(uploaded_file):
15
  docs = loader.load()
16
  documents.extend(docs)
17
  else:
18
- loader = TextLoader(tmp_location)
19
  docs = loader.load()
20
  documents.extend(docs)
21
  return documents
 
4
 
5
  def process_file(uploaded_file):
6
  # save the file temporarily
7
+ temp_file = "./temp/"+uploaded_file.path
8
  with open(temp_file, "wb") as file:
9
  file.write(uploaded_file.content)
10
  file_name = uploaded_file.name
 
15
  docs = loader.load()
16
  documents.extend(docs)
17
  else:
18
+ loader = TextLoader(temp_file)
19
  docs = loader.load()
20
  documents.extend(docs)
21
  return documents