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