CreatorPhan/Healthcare-QnA

This model using T5-large paradigm with 790M parameters. This training by alo-bac-si dataset, which crawl by me from alobacsi.com website.

from transformers import AutoTokenizer, T5ForConditionalGeneration


device = 'cpu'

model_path = "CreatorPhan/Healthcare-QnA"
model = T5ForConditionalGeneration.from_pretrained(model_path).to(device)
tokenizer = AutoTokenizer.from_pretrained(model_path)


prompt = "Công dụng của paracetamol là gì?"

tokens = tokenizer(prompt, return_tensors='pt').input_ids
output = model.generate(tokens.to(device), max_new_tokens=170)[0]
predict = tokenizer.decode(output, skip_special_tokens=True)
print(len(predict.split()))
print(predict)

Input:

Công dụng của paracetamol là gì?

Expected output:

Paracetamol còn có tên gọi khác là Acetaminophen là kháng sinh thuộc nhóm Fluoroquinolon. Paracetamol có tác dụng hạ sốt, giảm đau đầu. Thuốc dùng hạ sốt trong bệnh lý còi xương do mệt mỏi, suy nhược cơ thể, người cao tuổi có thể dùng được. Chào bạn, Thuốc Paracetamol còn có tên gọi khác là Paracetamol, là thuốc giảm đau giảm viêm sử dụng trong bệnh lý paracetamol, có tác dụng giảm đau, hạ sốt, giảm đau cho bệnh nhân bị chấn thương nặng. Nhưng mà, tác dụng phụ này không phải là do paracetamol gây nên, vì tùy vào cơ địa người bệnh, tác nhân gây bệnh, tình trạng bệnh nền, nếu dùng paracetamol không đúng hoặc quá liều có thể gây độc thận. Trường hợp của bạn do dùng paracetamol không đúng cách nên dẫn đến nguy cơ độc gan nên có thể bạn đã có tiền sử bị rối loạn chuyển hóa mỡ máu, có thể do lo lắng quá mức, bạn nên nhờ bác sĩ điều trị tư vấn trực tiếp khi dùng thuốc bạn nhé!