|
--- |
|
tags: |
|
- medical |
|
license: mit |
|
datasets: |
|
- Mohammed-Altaf/medical-instruction-120k |
|
language: |
|
- en |
|
library_name: transformers |
|
--- |
|
Please note that the chatbot is designed for research purposes only and is not intended for use in real medical settings. While the chatbot has been trained to provide accurate and helpful responses, it is not a substitute for professional medical advice, diagnosis, or treatment. The information provided by the chatbot should not be used to make medical decisions, and any health concerns should be addressed by a licensed healthcare provider. |
|
|
|
## Quickstart |
|
|
|
```python |
|
import torch |
|
from transformers import GPT2LMHeadModel, GPT2Tokenizer |
|
|
|
|
|
path = "Mohammed-Altaf/Medical-ChatBot" |
|
device = "cuda" if torch.cuda.is_available() else "cpu" |
|
tokenizer = GPT2Tokenizer.from_pretrained(path) |
|
model = GPT2LMHeadModel.from_pretrained(path).to(device) |
|
|
|
prompt_input = ( |
|
"The conversation between human and AI assistant.\n" |
|
"[|Human|] {input}\n" |
|
"[|AI|]" |
|
) |
|
sentence = prompt_input.format_map({'input': "what is Autism's disease?"}) |
|
inputs = tokenizer(sentence, return_tensors="pt").to(device) |
|
|
|
with torch.no_grad(): |
|
beam_output = model.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)) |
|
``` |
|
|
|
## Example Outputs |
|
``` |
|
The conversation between human and AI assistant. |
|
[|Human|] what is Autism's disease? |
|
[|AI|] Autism's disease, also known as Autism Spectrum Disorder (ASD), is a neurodevelopmental disorder that affects communication, social interaction, and behavior. |
|
``` |
|
|
|
``` |
|
The conversation between human and AI assistant. |
|
[|Human|] What are the side effects of autism? |
|
[|AI|] Autism is a neurodevelopmental disorder that affects communication, social interaction, and behavior. Some common side effects of autism include difficulty with speech and language, repetitive behaviors or routines, and sensory sensitivities. |
|
|
|
|
|
``` |
