Update audio2text/a2t.py

audio2text/a2t.py

@@ -10,18 +10,18 @@ class A2T:
     def __init__(self, mic):
         self.mic = mic
 
-    def __transcribe(self, inputs, task: str = None):
+    def __transcribe(self, inputs, task: str = None, lang: str = "english"):
         if inputs is  None:
             print("Inputs None")
 
-        transcribed_text = pipe(inputs, batch_size=BATCH_SIZE,)["text"]
+        transcribed_text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task, "language": lang}, return_timestamps=True)["text"]
         return transcribed_text
 
     def __preprocces(self, raw: np.ndarray, sampling_rate: int):
         chunk = raw.astype(np.float32) / 32678.0
 
-        if sampling_rate != 16000:
-            chunk = librosa.resample(chunk, orig_sr=sampling_rate, target_sr=16000)
+        # if sampling_rate != 16000:
+        #     chunk = librosa.resample(chunk, orig_sr=sampling_rate, target_sr=16000)
 
         # chunk = chunk[:16000*LIMIT]