File size: 1,391 Bytes
5403e87 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 | import argparse
import os
import warnings
import subprocess
warnings.filterwarnings("ignore")
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--gradio", action="store_true", help="Launch gradio UI")
parser.add_argument("--server", action="store_true", help="Launch flask server")
parser.add_argument("--test", action="store_true", help="Launch test")
parser.add_argument("--infer", help="Path to audio file for infernace")
parser.add_argument("--audio_location", help="Path to audio files")
args = parser.parse_args()
if args.gradio:
subprocess.run(["python", "src/gradio_app.py"])
if args.infer:
if not os.path.exists(args.infer):
print(f"File not found: {args.infer}")
return
from inference_file import inference
result = inference(args.infer)
print(result)
if args.test:
if args.audio_location:
if not os.path.exists(args.audio_location):
print(f"Folder not found: {args.audio_location}")
return
os.environ["AUDIO_LOCATION"] = args.audio_location
subprocess.run(["python", "src/model_test.py"])
else:
print("Missings option --audio_location")
if args.server:
subprocess.run(["python", "src/flask_server.py"])
if __name__ == "__main__":
main()
|