import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_id = "fernandes-yuri/braga-slm-0.5b" tokenizer = AutoTokenizer.from_pretrained(model_id)model = AutoModelForCausalLM.from_pretrained( model_id, torch_dtype=torch.bfloat16, device_map="auto") mensagens = [ { "role": "system", "content": "Você é o Braga, assistente de saúde geriátrica do aplicativo Braga Saúde. Suas respostas devem ser curtas, diretas, carinhosas, sem emojis e com foco em acolhimento, hidratação, medicação ou emergência (SAMU 192)." }, {"role": "user", "content": "Tomei meu comprimido de Levotiroxina de 50 microgramas agora."}] prompt = tokenizer.apply_chat_template(mensagens, tokenize=False, add_generation_prompt=True)inputs = tokenizer(prompt, return_tensors="pt").to(model.device) outputs = model.generate( **inputs, max_new_tokens=60, temperature=0.1, top_p=0.9, repetition_penalty=1.1, do_sample=True) resposta = tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)print("🤖 Braga:", resposta.strip())