bugfix

app.py

@@ -57,7 +57,7 @@ if __name__ == "__main__":
         )
         # Important: don't forget to set `normalize_logits=True` to obtain normalized probabilities (i.e. sum(p) = 1)
         transition_scores = model.compute_transition_scores(outputs.sequences, outputs.scores, normalize_logits=True)
-        transition_proba = np.exp(transition_scores)
+        transition_proba = np.exp(transition_scores[0])
         # We only have scores for the generated tokens, so pop out the prompt tokens
         input_length = 1 if model.config.is_encoder_decoder else inputs.input_ids.shape[1]
         generated_ids = outputs.sequences[:, input_length:]