jiuuee commited on
Commit
a031984
·
verified ·
1 Parent(s): dd54457

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
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
- input_text = [input_text]
 
97
  generation_args = {
98
- "max_new_tokens": 300,
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