Tonic commited on
Commit
d181372
·
1 Parent(s): a9a2a7f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,10 +24,11 @@ def save_image(image_file, upload_dir: str) -> str:
24
  Path(upload_dir).mkdir(parents=True, exist_ok=True)
25
  filename = secrets.token_hex(10) + Path(image_file.name).suffix
26
  file_path = Path(upload_dir) / filename
27
- with open(image_file, "rb") as f_input, open(file_path, "wb") as f_output:
28
- f_output.write(f_input.read())
29
  return str(file_path)
30
 
 
31
  def clean_response(response: str) -> str:
32
  response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
33
  return response
 
24
  Path(upload_dir).mkdir(parents=True, exist_ok=True)
25
  filename = secrets.token_hex(10) + Path(image_file.name).suffix
26
  file_path = Path(upload_dir) / filename
27
+ with open(file_path, "wb") as f_output:
28
+ f_output.write(image_file.read())
29
  return str(file_path)
30
 
31
+
32
  def clean_response(response: str) -> str:
33
  response = re.sub(r'<ref>(.*?)</ref>(?:<box>.*?</box>)*(?:<quad>.*?</quad>)*', r'\1', response).strip()
34
  return response