Update README.md

README.md

@@ -45,7 +45,8 @@ def ask_cosmosage(question):
     tokenizer.encode("ASSISTANT:", add_special_tokens=False, return_tensors="pt")
     ], dim=-1).to(device)
     generated_ids = model.generate(input_ids, max_length=input_ids.shape[1] + 1000, do_sample=True)
-    return tokenizer.decode(generated_ids[0], skip_special_tokens=True)```
+    return tokenizer.decode(generated_ids[0], skip_special_tokens=True)
+```
 
 ## Comparison to cosmosage_v1