Spaces:
Runtime error
Runtime error
Commit
·
6a3f9b4
1
Parent(s):
3dc3903
whisper llama
Browse files- whisper_app.py +1 -1
whisper_app.py
CHANGED
|
@@ -40,7 +40,6 @@ class WHISPERModel:
|
|
| 40 |
raise("Unable to reach for URL {}".format(audio_path))
|
| 41 |
else:
|
| 42 |
audio = self.read_audio(audio_path)
|
| 43 |
-
if audio :
|
| 44 |
conv_language = self.get_info(audio)
|
| 45 |
if conv_language !='en':
|
| 46 |
res = self.model.transcribe(audio,task='translate')
|
|
@@ -50,6 +49,7 @@ class WHISPERModel:
|
|
| 50 |
res = self.model.transcribe(audio)
|
| 51 |
audio_duration = audio.shape[0] / SAMPLE_RATE
|
| 52 |
text_data['text'] = res['text']
|
|
|
|
| 53 |
text_data['duration'] = audio_duration
|
| 54 |
text_data['language'] = conv_language
|
| 55 |
return text_data
|
|
|
|
| 40 |
raise("Unable to reach for URL {}".format(audio_path))
|
| 41 |
else:
|
| 42 |
audio = self.read_audio(audio_path)
|
|
|
|
| 43 |
conv_language = self.get_info(audio)
|
| 44 |
if conv_language !='en':
|
| 45 |
res = self.model.transcribe(audio,task='translate')
|
|
|
|
| 49 |
res = self.model.transcribe(audio)
|
| 50 |
audio_duration = audio.shape[0] / SAMPLE_RATE
|
| 51 |
text_data['text'] = res['text']
|
| 52 |
+
|
| 53 |
text_data['duration'] = audio_duration
|
| 54 |
text_data['language'] = conv_language
|
| 55 |
return text_data
|