testspace / src /rp_handler.py
StormblessedKal's picture
docker for runpod
e1412bc
raw
history blame
915 Bytes
"""
rp_handler.py for runpod worker
rp_debugger:
- Utility that provides additional debugging information.
The handler must be called with --rp_debugger flag to enable it.
"""
import base64
import tempfile
from rp_schema import INPUT_VALIDATIONS
from runpod.serverless.utils import download_files_from_urls, rp_cleanup, rp_debugger
from runpod.serverless.utils.rp_validator import validate
import runpod
import predict
MODEL = predict.Predictor()
MODEL.setup()
@rp_debugger.FunctionTimer
def run_voice_clone_job(job):
job_input = job['input']
method_type = job_input['method_type']
assert method_type in ["create_voice","voice_clone","voice_clone_with_emotions","voice_clone_with_multi_lang"]
s3_url = job_input['s3_url']
passage = job_input['passage']
processed_urls = MODEL.predict(s3_url,passage)
return processed_urls
runpod.serverless.start({"handler": run_voice_clone_job})