Update README.md

README.md

@@ -106,30 +106,28 @@ Here is how to use this model:
 
 ```python
 import torch
-import transformers
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
 
-input_text = "Maria y Miguel no tienen ningún "
-model = "projecte-aina/aguila-7b"
-tokenizer = AutoTokenizer.from_pretrained(model)
+input_text = "El mercat del barri és fantàstic, hi pots trobar"
 
-pipeline = transformers.pipeline(
+model_id  = "projecte-aina/aguila-7b"
+tokenizer = AutoTokenizer.from_pretrained(model_id)
+generator = pipeline(
     "text-generation",
-    model=model,
+    model=model_id,
     tokenizer=tokenizer,
     torch_dtype=torch.bfloat16,
     trust_remote_code=True,
     device_map="auto",
 )
-generation = pipeline(
+generation = generator(
     input_text,
-    max_length=200,
     do_sample=True,
     top_k=10,
     eos_token_id=tokenizer.eos_token_id,
 )
 
-print(f"Result: {generation['generated_text']}")
+print(f"Result: {generation[0]['generated_text']}")
 ```
 
 ## Limitations and bias