jebin2 commited on
Commit
e6e7958
·
1 Parent(s): 225e0e7

added foooter

Browse files
comic_panel_extractor/annorator_server.py CHANGED
@@ -353,4 +353,5 @@ async def upload_image(file: UploadFile = File(...)):
353
  file_path = os.path.join(IMAGE_ROOT, "train", file.filename)
354
  with open(file_path, "wb") as f:
355
  f.write(await file.read())
 
356
  return {"message": f"Uploaded {file.filename} to train set"}
 
353
  file_path = os.path.join(IMAGE_ROOT, "train", file.filename)
354
  with open(file_path, "wb") as f:
355
  f.write(await file.read())
356
+ shutil.copy(file_path, f'{Config.IMAGE_SOURCE_PATH}/{file.filename}')
357
  return {"message": f"Uploaded {file.filename} to train set"}