Update README.md

README.md

@@ -80,10 +80,11 @@ print(summary)
 
 The model currently should close its summary with a `</CARD_SUMMARY>` (cooking some more with this...), so you can also use this as a stopping criterion when using `pipeline` inference. 
 
-```
+```python
 from transformers import pipeline, StoppingCriteria, StoppingCriteriaList
 import torch
 
+
 class StopOnTokens(StoppingCriteria):
     def __init__(self, tokenizer, stop_token_ids):
         self.stop_token_ids = stop_token_ids
@@ -97,6 +98,7 @@ class StopOnTokens(StoppingCriteria):
                 return True
         return False
 
+
 # Initialize pipeline
 pipe = pipeline("text-generation", "davanstrien/Smol-Hub-tldr")
 tokenizer = pipe.tokenizer