Spaces:
mskov
/
Runtime error

mskov commited on
Commit
9cb6dfc
·
1 Parent(s): 1f73fee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -47,7 +47,8 @@ def inference(audio):
47
  ### code
48
  input_ids = tokenizer(result.text, return_tensors="pt").input_ids
49
  print("inputs ", input_ids)
50
- generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True, length_penalty=0)
 
51
  print("outputs generated ", generated_outputs[0])
52
  # only use id's that were generated
53
  # gen_sequences has shape [3, 15]
 
47
  ### code
48
  input_ids = tokenizer(result.text, return_tensors="pt").input_ids
49
  print("inputs ", input_ids)
50
+ # length penalty for gpt2.generate???
51
+ generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True)
52
  print("outputs generated ", generated_outputs[0])
53
  # only use id's that were generated
54
  # gen_sequences has shape [3, 15]