Update README.md

README.md

@@ -22,6 +22,7 @@ from transformers import pipeline, StoppingCriteriaList, StoppingCriteria
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+
 # (Optional) - define a stopping criteria
 # We ideally want the model to stop generate once the response from the Bot is generated
 class StopOnTokenCriteria(StoppingCriteria):
@@ -43,7 +44,12 @@ pipe = pipeline(
 text = "I like to eat ice cream in the summer."
 prompt = f"<|endoftext|><s>User: Översätt till Svenska från Engelska\n{text}<s>Bot:"
 
-response = pipe(prompt, max_length=768, stopping_criteria=StoppingCriteriaList([stop_on_token_criteria])))
+response = pipe(
+    prompt,
+    max_length=768,
+    stopping_criteria=StoppingCriteriaList([stop_on_token_criteria])
+)
+
 print(response[0]["generated_text"].split("<s>Bot: ")[-1])
 ```
 ```python