jxtan commited on
Commit
0913403
·
1 Parent(s): a1ec192

Bug fix server.py invalid job inputs

Browse files
Files changed (1) hide show
  1. server.py +4 -3
server.py CHANGED
@@ -68,9 +68,10 @@ def run_t2tt(input_text: str, source_language: str, target_language: str) -> str
68
  import runpod
69
 
70
  def runpod_handler(job):
71
- input_text = job["input_text"]
72
- source_language = job["source_language"]
73
- target_language = job["target_language"]
 
74
  return run_t2tt(input_text, source_language, target_language)
75
 
76
  runpod.serverless.start({"handler": runpod_handler})
 
68
  import runpod
69
 
70
  def runpod_handler(job):
71
+ job_input = job['input']
72
+ input_text = job_input["input_text"]
73
+ source_language = job_input["source_language"]
74
+ target_language = job_input["target_language"]
75
  return run_t2tt(input_text, source_language, target_language)
76
 
77
  runpod.serverless.start({"handler": runpod_handler})