easxtn commited on
Commit
af85530
1 Parent(s): 8ca6683

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def text_generation(text):
30
  }
31
 
32
  text_gen= pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code = True)
33
- response = text_gen(messages, generation_args)
34
  return response[0]["generated_text"]
35
 
36
  def text_to_speech(text):
 
30
  }
31
 
32
  text_gen= pipeline("text-generation", model=model, tokenizer=tokenizer, trust_remote_code = True)
33
+ response = text_gen(messages, **generation_args)
34
  return response[0]["generated_text"]
35
 
36
  def text_to_speech(text):