Modelpublic

Jnaranjo/qwen-1.8b-ecuador-qa

sourceHugging Faceapache-2.0updated 1y agoView on Hugging Face

0likes15downloads

Model Card

Qwen-1.8B Ecuador QA

Modelo fine-tuneado de Qwen-1.8B para responder preguntas en español sobre Ecuador.

Descripción

—Modelo Base: Qwen/Qwen-1_8B
—Dataset: Jnaranjo/EcuadorWeb-QA (10,597 ejemplos)
—Idioma: Español (contexto ecuatoriano)

Uso

Con Transformers

python

from transformers import AutoModelForCausalLM, AutoTokenizer

model = AutoModelForCausalLM.from_pretrained("Jnaranjo/qwen-1.8b-ecuador-qa", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("Jnaranjo/qwen-1.8b-ecuador-qa", trust_remote_code=True)

if tokenizer.pad_token is None:
    tokenizer.pad_token = tokenizer.eos_token

pregunta = "¿Dónde se encuentra el estudio 108 Yoga?"
prompt = f"Question: {pregunta}\nAnswer:"
inputs = tokenizer(prompt, return_tensors="pt", truncation=True, max_length=512)
outputs = model.generate(**inputs, max_new_tokens=256, temperature=0.7, do_sample=True)
respuesta = tokenizer.decode(outputs[0], skip_special_tokens=True)[len(prompt):].strip()
print(f"Respuesta: {respuesta}")