Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import soundfile as sf
+from scipy import signal
 import numpy as np
 import torch, torchaudio
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
@@ -15,6 +16,7 @@ processor_is = Wav2Vec2Processor.from_pretrained(MODEL_IS)
 model_fo = Wav2Vec2ForCTC.from_pretrained(MODEL_FO).to(device)
 processor_fo = Wav2Vec2Processor.from_pretrained(MODEL_FO) 
 
+
 def readwav(a_f):
     wav, sr = sf.read(a_f, dtype=np.float32)
     if len(wav.shape) == 2:
@@ -24,25 +26,23 @@ def readwav(a_f):
         wav = signal.resample(wav, wlen)
     return wav
 
-def recis(audio_file):
+def recc(audio_file,model,processor):
     wav = readwav(audio_file)
     with torch.inference_mode():
-        input_values = processor_is(wav,sampling_rate=16000).input_values[0]
+        input_values = processor(wav,sampling_rate=16000).input_values[0]
         input_values = torch.tensor(input_values, device=device).unsqueeze(0)
-        logits = model_is(input_values).logits
+        logits = model(input_values).logits
         pred_ids = torch.argmax(logits, dim=-1)
-        xcp = processor_is.batch_decode(pred_ids)
-        return xcp
+        xcp = processor.batch_decode(pred_ids)
+        return xcp[0]
+
+    
+def recis(audio_file):
+    return recc(audio_file,model_is,processor_is)
 
 def recfo(audio_file):
-    wav = readwav(audio_file)
-    with torch.inference_mode():
-        input_values = processor_fo(wav,sampling_rate=16000).input_values[0]
-        input_values = torch.tensor(input_values, device=device).unsqueeze(0)
-        logits = model_fo(input_values).logits
-        pred_ids = torch.argmax(logits, dim=-1)
-        xcp = processor_fo.batch_decode(pred_ids)
-        return xcp
+    return recc(audio_file,model_fo,processor_fo)
+
 
 bl = gr.Blocks()
 with bl:
@@ -54,8 +54,9 @@ with bl:
     https://huggingface.co/carlosdanielhernandezmena/wav2vec2-large-xlsr-53-icelandic-ep10-1000h 
     or https://huggingface.co/carlosdanielhernandezmena/wav2vec2-large-xlsr-53-faroese-100h  
 
-    - For some reason, the huggingface 'Hosted inference API' on the model page does not work, but this does.  
+    - For some reason, the huggingface 'Hosted inference API' on the model page does not currently work, but this does.  
     - There is no language model (yet), so it can generate non-words.
+    - Send errors/bugs to [email protected]
     """
     )