Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
@app.function(
|
| 2 |
class InteractiveChat:
|
| 3 |
def __init__(self, model_name="openai/whisper-large", tts_choice="OpenVoice", **kwargs):
|
| 4 |
self.whisper_processor = WhisperProcessor.from_pretrained(model_name)
|
|
@@ -13,7 +13,7 @@ class InteractiveChat:
|
|
| 13 |
predicted_ids = self.whisper_model.generate(input_features)
|
| 14 |
transcription = self.whisper_processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
|
| 15 |
|
| 16 |
-
|
| 17 |
response = self.zephyr_pipeline(transcription, max_length=1000)[0]["generated_text"]
|
| 18 |
return transcription, response
|
| 19 |
|
|
@@ -50,4 +50,4 @@ with gr.Blocks() as demo:
|
|
| 50 |
input_data.change(lambda x, model: model.generate_response(x), inputs=[input_data, model_choice],
|
| 51 |
outputs=output_text)
|
| 52 |
input_data.change(lambda x, model: model.speak(x[1]), inputs=[output_text, model_choice],
|
| 53 |
-
outputs=None) # Speak the response
|
|
|
|
| 1 |
+
@app.function(
|
| 2 |
class InteractiveChat:
|
| 3 |
def __init__(self, model_name="openai/whisper-large", tts_choice="OpenVoice", **kwargs):
|
| 4 |
self.whisper_processor = WhisperProcessor.from_pretrained(model_name)
|
|
|
|
| 13 |
predicted_ids = self.whisper_model.generate(input_features)
|
| 14 |
transcription = self.whisper_processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
|
| 15 |
|
| 16 |
+
|
| 17 |
response = self.zephyr_pipeline(transcription, max_length=1000)[0]["generated_text"]
|
| 18 |
return transcription, response
|
| 19 |
|
|
|
|
| 50 |
input_data.change(lambda x, model: model.generate_response(x), inputs=[input_data, model_choice],
|
| 51 |
outputs=output_text)
|
| 52 |
input_data.change(lambda x, model: model.speak(x[1]), inputs=[output_text, model_choice],
|
| 53 |
+
outputs=None)) # Speak the response
|