somosnlp
/

gua-a

@@ -111,7 +111,7 @@ model = AutoModelForCausalLM.from_pretrained("somosnlp/gua-a", trust_remote_code
 input_text = "Sobre que trata el libro?"
 input_ids = tokenizer.encode(input_text)
-top_texts = model.document_extractor(query_outputs, k=5) # k: número de documentos recuperados
 print([tokenizer.decode(text, skip_special_tokens=True) for text in top_texts])
 ```

 input_text = "Sobre que trata el libro?"
 input_ids = tokenizer.encode(input_text)
+top_texts = model.document_extractor(input_ids, k=5) # k: número de documentos recuperados
 print([tokenizer.decode(text, skip_special_tokens=True) for text in top_texts])
 ```