Commit
·
17630db
1
Parent(s):
8efe78a
debug filename
Browse files- handler.py +3 -3
handler.py
CHANGED
|
@@ -245,9 +245,9 @@ class EndpointHandler:
|
|
| 245 |
w.write(inputs)
|
| 246 |
|
| 247 |
# audio_nparray = ffmpeg_load_audio('/tmp/myfile.tmp', sr=SAMPLE_RATE, mono=True, out_type=np.float32)
|
| 248 |
-
audio_nparray = load_audio("/tmp/myfile.tmp", sr=SAMPLE_RATE)
|
| 249 |
# clean up
|
| 250 |
-
os.remove("/tmp/myfile.tmp")
|
| 251 |
|
| 252 |
# audio_nparray = ffmpeg_read(inputs, SAMPLE_RATE)
|
| 253 |
# audio_tensor= torch.from_numpy(audio_nparray)
|
|
@@ -264,7 +264,7 @@ class EndpointHandler:
|
|
| 264 |
# 2. transcribe
|
| 265 |
logger.info("--------------- STARTING TRANSCRIPTION ------------------------")
|
| 266 |
# transcription = self.model.transcribe(audio_nparray, batch_size=batch_size, language=language)
|
| 267 |
-
segments, _ = self.model.transcribe("
|
| 268 |
if info:
|
| 269 |
# print(transcription["segments"][0:10000]) # before alignment
|
| 270 |
print(segments[0:10000]) # before alignment
|
|
|
|
| 245 |
w.write(inputs)
|
| 246 |
|
| 247 |
# audio_nparray = ffmpeg_load_audio('/tmp/myfile.tmp', sr=SAMPLE_RATE, mono=True, out_type=np.float32)
|
| 248 |
+
# audio_nparray = load_audio("/tmp/myfile.tmp", sr=SAMPLE_RATE)
|
| 249 |
# clean up
|
| 250 |
+
# os.remove("/tmp/myfile.tmp")
|
| 251 |
|
| 252 |
# audio_nparray = ffmpeg_read(inputs, SAMPLE_RATE)
|
| 253 |
# audio_tensor= torch.from_numpy(audio_nparray)
|
|
|
|
| 264 |
# 2. transcribe
|
| 265 |
logger.info("--------------- STARTING TRANSCRIPTION ------------------------")
|
| 266 |
# transcription = self.model.transcribe(audio_nparray, batch_size=batch_size, language=language)
|
| 267 |
+
segments, _ = self.model.transcribe("/tmp/myfile.tmp", beam_size=5)
|
| 268 |
if info:
|
| 269 |
# print(transcription["segments"][0:10000]) # before alignment
|
| 270 |
print(segments[0:10000]) # before alignment
|