feat: add code

index.html

@@ -131,17 +131,16 @@
       </div>
 
       <div class="card">
-        <h2>Loading</h2>
-        <p>Depending on how the checkpoint is published, you can load via Transformers-native code or custom code.</p>
-        <div class="code"><code>from transformers import AutoModel, AutoTokenizer
+        <h2>Model usage (to update)</h2>
+        <p>Here is a quick example of how to use NTv3 models.</p>
+        <div class="code"><code>from transformers import pipeline
 
-model = AutoModel.from_pretrained(
-    "InstaDeepAI/ntv3_106M_7downsample_post_trained_1mb",
-    trust_remote_code=True
-)
-tokenizer = AutoTokenizer.from_pretrained(
-    "InstaDeepAI/ntv3_106M_7downsample_post_trained_1mb",
-    trust_remote_code=True
+pipe = pipeline(
+    task="ntv3-tracks",
+    model="InstaDeepAI/ntv3_106M_7downsample_post_trained_1mb",
+    trust_remote_code=True,
+    device="cuda",
+    torch_dtype=torch.bfloat16,
 )</code></div>
       </div>