## Aviso Legal Este chatbot foi projetado apenas para fins de pesquisa e não se destina ao uso em configurações médicas reais. Embora o chatbot tenha sido treinado para fornecer respostas precisas e úteis, ele não substitui o aconselhamento médico profissional, o diagnóstico ou o tratamento. As informações fornecidas pelo chatbot não devem ser usadas para tomar decisões médicas, e quaisquer preocupações com a saúde devem ser tratadas por um profissional de saúde licenciado. ## Início Rápido ```python import torch from transformers import GPT2LMHeadModel, GPT2Tokenizer caminho = "Mohammed-Altaf/Medical-ChatBot" dispositivo = "cuda" if torch.cuda.is_available() else "cpu" tokenizer = GPT2Tokenizer.from_pretrained(caminho) modelo = GPT2LMHeadModel.from_pretrained(caminho).to(dispositivo) entrada_prompt = ( "A conversa entre humano e assistente de IA.\n" "[|Humano|] {input}\n" "[|IA|]" ) frase = entrada_prompt.format_map({'input': "o que é a doença de Parkinson?"}) inputs = tokenizer(frase, return_tensors="pt").to(dispositivo) with torch.no_grad(): beam_output = modelo.generate(**inputs, min_new_tokens=1, max_length=512, num_beams=3, repetition_penalty=1.2, early_stopping=True, eos_token_id=198 ) print(tokenizer.decode(beam_output[0], skip_special_tokens=True)) ``` ## Exemplos de Saída ``` A conversa entre humano e assistente de IA. [|Humano|] o que é a doença de Parkinson? [|IA|] A doença de Parkinson é um distúrbio neurodegenerativo que afeta o movimento. É causada pela perda de células produtoras de dopamina no cérebro. ``` ``` A conversa entre humano e assistente de IA. [|Humano|] que tipo de mel é melhor para uma tosse ruim de covid? [|IA|] O mel de Manuka mostrou ter propriedades anti-inflamatórias e antibacterianas que podem ajudar a aliviar os sintomas de uma tosse ruim de covid. ``` ---