Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import numpy as np
 import torch
 from datasets import load_dataset
+import torchaudio
 
 from transformers import pipeline
 from transformers import BarkModel, BarkProcessor
@@ -23,8 +24,9 @@ bark_model.to(device)
 
 def translate(audio):
     sr, y = audio
-    # if sr != 16000:
-    #     y = torchaudio.resample(y, sr, 16000)
+    if sr != 16000:
+        resampler = torchaudio.transforms.Resample(sr, 16000, dtype=y.dtype)
+        y = resampler(y)
     y = y.astype(np.float32)
     y /= np.max(np.abs(y))
     inputs = asr_processor(y, sampling_rate=16000, return_tensors="pt")