Medical-ChatBot / README.md
DHEIVER's picture
Update README.md
7f2597b verified
|
raw
history blame
2.13 kB

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

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.