model control

app.py

@@ -388,15 +388,18 @@ class WhisperTranscriber:
             language_detection_segments=1,
             task="translate" if translate else "transcribe",
         )
-        transcribe_model = whisper
         if batch_size > 1:
-            transcribe_model = batched_whisper
-        # Use batched inference for better performance
-        segments, transcript_info = transcribe_model.transcribe(
-            audio_path, 
-            #batch_size=batch_size, 
-            **options
-        )
+            # Use batched inference for better performance
+            segments, transcript_info = batched_whisper.transcribe(
+                audio_path, 
+                batch_size=batch_size, 
+                **options
+            )
+        else:
+            segments, transcript_info = whisper.transcribe(
+                audio_path, 
+                **options
+            )
         segments = list(segments)
         
         detected_language = transcript_info.language