Update README.md

README.md

@@ -40,6 +40,7 @@ This model is trained with the following objectives:
 ## Quick Start
 
 ```python
+from datasets import load_dataset
 from transformers import AutoModel, AutoProcessor
 import torch
 
@@ -48,12 +49,9 @@ model = AutoModel.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code
 processor = AutoProcessor.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code=True)
 model.eval()
 
-# Example: Predict word from swipe path
-from datasets import load_dataset
-
 # Load sample
-dataset = load_dataset("futo-org/swipe.futo.org", split="test[:1]")
-item = dataset[0]
+dataset = load_dataset("futo-org/swipe.futo.org", split="test[:10]")
+item = dataset[4]
 
 # Preprocess swipe path using processor methods
 # 1. Normalize timestamps (x,y already normalized in futo dataset)
@@ -148,7 +146,7 @@ mask_token_id = processor.tokenizer.mask_token_id
 masked_ids = [mask_token_id if cid != 0 else 0 for cid in char_ids]
 inputs["input_ids"] = torch.tensor([masked_ids], dtype=torch.long)
 
-# Predict from path only - achieves 94% character accuracy
+# Predict from path only
 ```
 
 ### Length Prediction