Update app.py
Browse files
app.py
CHANGED
@@ -44,9 +44,9 @@ def inference(audio):
|
|
44 |
|
45 |
### code
|
46 |
input_ids = tokenizer(result.text, return_tensors="pt").input_ids
|
47 |
-
|
48 |
generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True)
|
49 |
-
|
50 |
# only use id's that were generated
|
51 |
# gen_sequences has shape [3, 15]
|
52 |
gen_sequences = generated_outputs.sequences[:, input_ids.shape[-1]:]
|
|
|
44 |
|
45 |
### code
|
46 |
input_ids = tokenizer(result.text, return_tensors="pt").input_ids
|
47 |
+
print("inputs ", input_ids)
|
48 |
generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True)
|
49 |
+
print("outputs generated ", generated_outputs)
|
50 |
# only use id's that were generated
|
51 |
# gen_sequences has shape [3, 15]
|
52 |
gen_sequences = generated_outputs.sequences[:, input_ids.shape[-1]:]
|