Update README.md

README.md

@@ -105,6 +105,16 @@ def predict_memory_type(text):
     return model.config.id2label[predicted_id]
 
 predict_memory_type("Without a map, I navigated the winding back roads to reach my childhood home.")
+```
+or via huggingface pipeline
+```python
+# Use a pipeline as a high-level helper
+from transformers import pipeline
+import torch
+device = 0 if torch.cuda.is_available() else -1  # 0 = GPU, -1 = CPU
+pipe = pipeline("text-classification", model="DimitriosPanagoulias/MemoryBERT", device=device)
+pipe("I remember the long walk to my childhood school.")
+
 ```
 
 ## Citation