adjusted tensor shape

app.py

@@ -66,7 +66,7 @@ if __name__ == "__main__":
         if model.config.is_encoder_decoder:
             highlighted_out = []
         else:
-            input_tokens = tokenizer.convert_ids_to_tokens(inputs.input_ids)
+            input_tokens = tokenizer.convert_ids_to_tokens(inputs.input_ids[0])
             highlighted_out = [(token.replace("▁", " "), None) for token in input_tokens]
         # Get the (decoded_token, label) pairs for the generated tokens
         for token, proba in zip(generated_tokens, transition_proba[0]):