Yurii Paniv commited on
Commit
ce406e3
1 Parent(s): 7037773

Change gen parameters

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def transcribe(audio, selected_voice, history):
40
  past_user_inputs = [i[0] for i in history]
41
  generated_responses = [i[1] for i in history]
42
  next_output_length = len(tokenizer.encode("".join(generated_responses + past_user_inputs))) + 60
43
- response = conv(Conversation(text, past_user_inputs, generated_responses), max_length=next_output_length, penalty_alpha=0.6, top_k=4)
44
  response = response.generated_responses[-1]
45
  history.append((text, response))
46
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp:
 
40
  past_user_inputs = [i[0] for i in history]
41
  generated_responses = [i[1] for i in history]
42
  next_output_length = len(tokenizer.encode("".join(generated_responses + past_user_inputs))) + 60
43
+ response = conv(Conversation(text, past_user_inputs, generated_responses), max_length=next_output_length)
44
  response = response.generated_responses[-1]
45
  history.append((text, response))
46
  with tempfile.NamedTemporaryFile(suffix=".wav", delete=False) as fp: