Update app.py
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def transcribe_and_generate_audio(audio):
|
|
27 |
asr_output = transcriber({"sampling_rate": sr, "raw": y})["text"]
|
28 |
|
29 |
# Generate text based on ASR output
|
30 |
-
generated_text = generator(
|
31 |
|
32 |
# Generate audio from text using TTS model
|
33 |
inputs = tokenizer_tts(text=generated_text, return_tensors="pt")
|
|
|
27 |
asr_output = transcriber({"sampling_rate": sr, "raw": y})["text"]
|
28 |
|
29 |
# Generate text based on ASR output
|
30 |
+
generated_text = generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
|
31 |
|
32 |
# Generate audio from text using TTS model
|
33 |
inputs = tokenizer_tts(text=generated_text, return_tensors="pt")
|