Update README.md

README.md

@@ -39,15 +39,21 @@ The model achieves a loss rate of 1.4608 on the validation dataset after one epo
 See the snippet below for usage with Transformers:
 
 ```python
->>> import transformers
->>> import torch
+import transformers
+import torch
 
->>> model_id = "meta-llama/Meta-Llama-3-8B"
+model_id = "catallama/CataLlama-v0.1-Base"
 
->>> pipeline = transformers.pipeline(
-    "text-generation", model=model_id, model_kwargs={"torch_dtype": torch.bfloat16}, device_map="auto"
+pipeline = transformers.pipeline(
+    "text-generation",
+    model=model_id,
+    model_kwargs={"torch_dtype": torch.bfloat16},
+    device_map="auto",
 )
->>> pipeline("Hey how are you doing today?")
+
+outputs = pipeline("Ei com estàs avui?")
+
+print(outputs[0]["generated_text"][len(prompt):])
 ```