keess commited on
Commit
3e6a512
1 Parent(s): e5983bf

- add custom endpoint handler

Browse files
Files changed (2) hide show
  1. requirements.txt +0 -6
  2. test-handler.py +14 -0
requirements.txt CHANGED
@@ -1,11 +1,5 @@
1
- soundfile
2
  transformers
3
  torch
4
- sentencepiece
5
- librosa
6
  torchaudio
7
- pyctcdecode
8
- onnx
9
- onnxruntime
10
  pandas
11
  datasets
 
 
1
  transformers
2
  torch
 
 
3
  torchaudio
 
 
 
4
  pandas
5
  datasets
test-handler.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from handler import EndpointHandler
2
+ from pathlib import Path
3
+
4
+
5
+ p = Path(__file__).with_name('pytorch_model.bin')
6
+ filename = p.absolute()
7
+ my_handler = EndpointHandler(path=filename)
8
+
9
+ payload = {"inputs": "contact.wav"}
10
+
11
+ transcription=my_handler(payload)
12
+
13
+ print("here is the transcription")
14
+ print(transcription)