mskov commited on
Commit
3f8c47b
·
1 Parent(s): 757d6df

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)[:-4]
65
- print("outputs generated ", generated_outputs[0])
66
  # only use id's that were generated
67
  # gen_sequences has shape [3, 15]
68
  gen_sequences = generated_outputs.sequences[:, input_ids.shape[-1]:]
 
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)[:-4]
65
+ # print("outputs generated ", generated_outputs[0])
66
  # only use id's that were generated
67
  # gen_sequences has shape [3, 15]
68
  gen_sequences = generated_outputs.sequences[:, input_ids.shape[-1]:]