Spaces:
Running
Running
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"}
|