diff --git a/app.py b/app.py index bee9285..33e6e78 100644 --- a/app.py +++ b/app.py @@ -133,7 +133,7 @@ with gr.Blocks() as demo: isPipelineEnabledCheckbox.change(updateButton, inputs=[isPipelineEnabledCheckbox], outputs=refineTextBtn) saveFileCheckbox.change(updateTextbox, inputs=saveFileCheckbox, outputs=filename) - recognizeBtn.click(FasterWhisper.recognize, outputs=[recognizedText], inputs=[fastWhisperModel, audioFile, beamSize, vadFilter, minSilenceDurationMs, speechPadMs, temp0, temp1, temp2, wordTimestamps, noSpeechThreshold, conditionOnPreviousText]) + recognizeBtn.click(FasterWhisper().recognize, outputs=[recognizedText], inputs=[fastWhisperModel, audioFile, beamSize, vadFilter, minSilenceDurationMs, speechPadMs, temp0, temp1, temp2, wordTimestamps, noSpeechThreshold, conditionOnPreviousText]) # Если пайплайн включен то тогда делаем автоматически # автоматический пайплайн diff --git a/services/fasterWhisper.py b/services/fasterWhisper.py index 9072b53..ddcf22e 100644 --- a/services/fasterWhisper.py +++ b/services/fasterWhisper.py @@ -1,9 +1,6 @@ from faster_whisper import WhisperModel class FasterWhisper: - def __init__(self): - pass - def recognize(self, model, audioFile, beamSize, vadFilter, minSilenceDurationMs, speechPadMs, temp0, temp1, temp2, wordTimestamps, noSpeechThreshold, conditionOnPreviousText): model = WhisperModel(model, device='cuda', compute_type='float16') # Задаем модель