Update README.md

README.md

@@ -23,7 +23,7 @@ import torch
 from peft import PeftModel, PeftConfig
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-peft_model_id = "hackathon-somos-nlp-2023/bertin-gpt-j-6B-es-finetuned-salpaca"
+peft_model_id = "hackathon-somos-nlp-2023/SalpiBloom-1b1"
 config = PeftConfig.from_pretrained(peft_model_id)
 model = AutoModelForCausalLM.from_pretrained(config.base_model_name_or_path, return_dict=True, load_in_8bit=True, device_map='auto')
 # tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
@@ -40,8 +40,7 @@ def gen_conversation(text):
 
   print('\n\n', tokenizer.decode(output_tokens[0], skip_special_tokens=False))
 
-text = "hola"
-
+text = "Redacta un cuento corto"
 gen_conversation(text)
 ```