Update app.py
Browse files
app.py
CHANGED
@@ -93,17 +93,15 @@ def transcribe(audio_filepath):
|
|
93 |
|
94 |
# Function to generate text using LLM
|
95 |
def generate_text(input_text):
|
96 |
-
|
|
|
97 |
generation_args = {
|
98 |
-
"max_new_tokens":
|
99 |
"return_full_text": True,
|
100 |
"temperature": 0.0,
|
101 |
"do_sample": False,
|
102 |
}
|
103 |
-
generated_text = pipe(
|
104 |
-
[{"role": "user", "content": input_text}],
|
105 |
-
**generation_args
|
106 |
-
)[0]["generated_text"]
|
107 |
return generated_text
|
108 |
|
109 |
# Function to convert text to speech using TTS
|
|
|
93 |
|
94 |
# Function to generate text using LLM
|
95 |
def generate_text(input_text):
|
96 |
+
messages=[{"role": "user", "content": input_text},
|
97 |
+
]
|
98 |
generation_args = {
|
99 |
+
"max_new_tokens": 200,
|
100 |
"return_full_text": True,
|
101 |
"temperature": 0.0,
|
102 |
"do_sample": False,
|
103 |
}
|
104 |
+
generated_text = pipe(messages, **generation_args)[0]["generated_text"]
|
|
|
|
|
|
|
105 |
return generated_text
|
106 |
|
107 |
# Function to convert text to speech using TTS
|