| ## 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. | |
| ``` | |
| --- | |