Update README.md

README.md

@@ -114,9 +114,11 @@ 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.")
-
 ```
-
+outputs:
+```bash
+[{'label': 'episodic', 'score': 0.9272529482841492}]
+```
 ## Citation
 
 You can cite either one or both of the following previous related work: