Update app.py
Browse files
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 |
-
|
|
|
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]
|