Update app.py

app.py

@@ -42,6 +42,7 @@ def visualize_transformer(model_name, sentence):
     
     if "t5" in model_name:
         model = AutoModelForSeq2SeqLM.from_pretrained(model_name, output_attentions=True)
+        sentence = "translate English to English: " + sentence
         inputs = tokenizer(sentence, return_tensors='pt')
     elif "gpt2" in model_name:
         model = GPT2Model.from_pretrained(model_name, output_attentions=True)
@@ -65,8 +66,8 @@ def visualize_transformer(model_name, sentence):
     plt.xticks(rotation=90)
     plt.yticks(rotation=0)
 
-    token_output = [f"{i}: \"{tok}\"" for i, tok in enumerate(tokens)]
-    token_output_str = "[\\n" + "\\n".join(token_output) + "\\n]"
+    token_output = [f"{i + 1}: \"{tok}\"" for i, tok in enumerate(tokens)]
+    token_output_str = "[\n" + "\n".join(token_output) + "\n]"
 
     model_info = MODEL_INFO.get(model_name, {})
     details = f"""