Penality commited on
Commit
08d16b1
·
verified ·
1 Parent(s): 324d50d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -152,14 +152,14 @@ def retrieve_document(query):
152
  print(response.content)
153
 
154
  if response.status_code == 200:
155
- with open(filename, "wb") as f:
156
  f.write(response.content)
157
 
158
  print(f"PDF received successfully: received_{filename}")
159
  else:
160
  print(f"Error: {response.status_code}, {response.json()}")
161
 
162
- return filename
163
 
164
  def clean_text(text):
165
  """Cleans extracted text for better processing by the model."""
 
152
  print(response.content)
153
 
154
  if response.status_code == 200:
155
+ with open(recieved_file, "wb") as f:
156
  f.write(response.content)
157
 
158
  print(f"PDF received successfully: received_{filename}")
159
  else:
160
  print(f"Error: {response.status_code}, {response.json()}")
161
 
162
+ return recieved_file
163
 
164
  def clean_text(text):
165
  """Cleans extracted text for better processing by the model."""