feat: improve main page

index.html

@@ -337,7 +337,11 @@ tok = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 batch = tok(["ATCGNATCG", "ACGT"], add_special_tokens=False, padding=True, pad_to_multiple_of=128, return_tensors="pt")
 
 # Run model
-out = model(**batch, output_hidden_states=True, output_attentions=True)
+out = model(
+  **batch,
+  output_hidden_states=True,
+  output_attentions=True
+)
 
 # Print output shapes
 print(out.logits.shape)       # (B, L, V = 11)