فهرست منبع

Merge pull request #58 from jaggzh/detect-input-device

Auto-detection of input device index added
Kolja Beigel 11 ماه پیش
والد
کامیت
736e8e5084
1فایلهای تغییر یافته به همراه4 افزوده شده و 1 حذف شده
  1. 4 1
      RealtimeSTT/audio_recorder.py

+ 4 - 1
RealtimeSTT/audio_recorder.py

@@ -90,7 +90,7 @@ class AudioToTextRecorder:
                  model: str = INIT_MODEL_TRANSCRIPTION,
                  language: str = "",
                  compute_type: str = "default",
-                 input_device_index: int = 0,
+                 input_device_index: int = None,
                  gpu_device_index: Union[int, List[int]] = 0,
                  device: str = "cuda",
                  on_recording_start=None,
@@ -708,6 +708,9 @@ class AudioToTextRecorder:
         """
         try:
             audio_interface = pyaudio.PyAudio()
+            if input_device_index is None:
+                default_device = audio_interface.get_default_input_device_info()
+                input_device_index = default_device['index']
             stream = audio_interface.open(
                 rate=sample_rate,
                 format=pyaudio.paInt16,