jt-whisper / test.py
Leslie Jones-Dove
add custom handler
f0386c6
from handler import EndpointHandler
from datasets import load_dataset
# init handler
my_handler = EndpointHandler(path=".")
# print(type(sample["data"]))
prediction = my_handler({"inputs": "/Users/lesliejones-dove/Desktop/projects/Whisper/whisperx/whisperCustomHandler/Saffored_short.wav"})
# /Users/lesliejones-dove/Desktop/projects/Whisper/whisperx/whisperCustomHandler
print(prediction)