Update app.py
Browse files
app.py
CHANGED
|
@@ -60,14 +60,14 @@ def imageMode(image, question):
|
|
| 60 |
return "In progress"
|
| 61 |
|
| 62 |
def audioMode(audio):
|
| 63 |
-
print('---------type of audio--------------')
|
| 64 |
-
sampling_rate = audio[0]
|
| 65 |
audio_array = audio[1]
|
| 66 |
-
print(sampling_rate)
|
| 67 |
-
print(audio_array)
|
| 68 |
-
input_features = processor_audio(audio_array, sampling_rate=
|
| 69 |
predicted_ids = model_audio.generate(input_features)
|
| 70 |
-
transcription =
|
| 71 |
return transcription
|
| 72 |
|
| 73 |
|
|
|
|
| 60 |
return "In progress"
|
| 61 |
|
| 62 |
def audioMode(audio):
|
| 63 |
+
#print('---------type of audio--------------')
|
| 64 |
+
#sampling_rate = audio[0]
|
| 65 |
audio_array = audio[1]
|
| 66 |
+
#print(sampling_rate)
|
| 67 |
+
#print(audio_array)
|
| 68 |
+
input_features = processor_audio(audio_array, sampling_rate=16000, return_tensors="pt").input_features
|
| 69 |
predicted_ids = model_audio.generate(input_features)
|
| 70 |
+
transcription = processor_audio.batch_decode(predicted_ids, skip_special_tokens=True)
|
| 71 |
return transcription
|
| 72 |
|
| 73 |
|