How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="HaiderSultanArc/UnaniBERT")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("HaiderSultanArc/UnaniBERT")
model = AutoModelForSequenceClassification.from_pretrained("HaiderSultanArc/UnaniBERT")
Quick Links

Model Details

Fine-tuned variant of BioBERT, for text classification task on user's input with symptoms and/or causes to predict possible disease, aimed towards Decision Support System for Unani Medicine Practitioners.

Model Description

Uses

Input a sentence with Symptoms and/or causes to predict possible diseases.

Downloads last month
4
Safetensors
Model size
0.1B params
Tensor type
I64
·
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Space using HaiderSultanArc/UnaniBERT 1

Paper for HaiderSultanArc/UnaniBERT