mskov commited on
Commit
ea1ad1e
·
1 Parent(s): 5b7e87f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -62,7 +62,7 @@ def inference(audio, state=""):
62
  # length penalty for gpt2.generate???
63
  #Prompt
64
  #generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True, max_length=4)
65
- output = model.generate(input_ids, max_length=5, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=5)
66
  print("output ", output)
67
  #outputs = [output[-4:] for output in output.tolist()]
68
  # print("outputs generated ", generated_outputs[0])
 
62
  # length penalty for gpt2.generate???
63
  #Prompt
64
  #generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True, max_length=4)
65
+ output = gpt2.generate(input_ids, max_length=5, do_sample=True, top_k=50, top_p=0.95, num_return_sequences=5)
66
  print("output ", output)
67
  #outputs = [output[-4:] for output in output.tolist()]
68
  # print("outputs generated ", generated_outputs[0])