Update app.py

app.py

@@ -46,7 +46,7 @@ def inference(audio):
     input_ids = tokenizer(result.text, return_tensors="pt").input_ids
     print("inputs ", input_ids)
     generated_outputs = gpt2.generate(input_ids, do_sample=True, num_return_sequences=3, output_scores=True)
-    print("outputs generated ", generated_outputs)
+    print("outputs generated ", generated_outputs[0])
     # only use id's that were generated
     # gen_sequences has shape [3, 15]
     gen_sequences = generated_outputs.sequences[:, input_ids.shape[-1]:]
@@ -58,7 +58,7 @@ def inference(audio):
     # now we need to collect the probability of the generated token
     # we need to add a dummy dim in the end to make gather work
     gen_probs = torch.gather(probs, 2, gen_sequences[:, :, None]).squeeze(-1)
-    
+    print("gen probs result: ", gen_probs)
     # now we can do all kinds of things with the probs
     
     # 1) the probs that exactly those sequences are generated again