pbotsaris commited on
Commit
aa06a1d
·
1 Parent(s): 3ce0400

added a few more debug returns

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -114,7 +114,7 @@ class EndpointHandler:
114
  shape += ":" + str(v)
115
 
116
  base64_encoded_wav = base64.b64encode(wav_data).decode('utf-8')
117
- return [{"audio": base64_encoded_wav, "wav_len": w_len, "pred_len": p_len, "shape": shape}]
118
 
119
 
120
  if __name__ == "__main__":
 
114
  shape += ":" + str(v)
115
 
116
  base64_encoded_wav = base64.b64encode(wav_data).decode('utf-8')
117
+ return [{"audio": base64_encoded_wav, "wav_len": w_len, "pred_len": p_len, "shape": shape, "sr": sr, 'dtype': str(pred.dtype)}]
118
 
119
 
120
  if __name__ == "__main__":