Medical-ChatBot / README.md
DHEIVER's picture
Update README.md
7f2597b verified
|
raw
history blame
No virus
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
```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.
```
---