forked from svc-develop-team/so-vits-svc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
wav_upload.py
21 lines (20 loc) · 783 Bytes
/
wav_upload.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from google.colab import files
import shutil
import os
import argparse
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.add_argument("--type", type=str, required=True, help="type of file to upload")
args = parser.parse_args()
file_type = args.type
basepath = os.getcwd()
uploaded = files.upload()
assert(file_type in ['zip', 'audio'])
if file_type == "zip":
upload_path = "./upload/"
for filename in uploaded.keys():
shutil.move(os.path.join(basepath, filename), os.path.join(upload_path, "userzip.zip"))
elif file_type == "audio":
upload_path = "./raw/"
for filename in uploaded.keys():
shutil.move(os.path.join(basepath, filename), os.path.join(upload_path, filename))