is there a way to retrieve an array of the tags?

#41
by gueroverde - opened

when i tried to run in my local machine the response only contains 1 tag, but i can see in the inferent api that return around 8 results, so how can i configure to retrieve more than 1 ?

this is my script

Use a pipeline as a high-level helper

from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification

pipe = pipeline("text-classification", model="sid321axn/Bio_ClinicalBERT-finetuned-medicalcondition")

tokenizer = AutoTokenizer.from_pretrained("sid321axn/Bio_ClinicalBERT-finetuned-medicalcondition")
model = AutoModelForSequenceClassification.from_pretrained("sid321axn/Bio_ClinicalBERT-finetuned-medicalcondition")

print(pipe("Intracranial Pressure, Cerebral Edema, and Brain Bleeds,"))

and this is the response

[{'label': 'High Blood Pressure', 'score': 0.9891742467880249}]

but i tried in the card seems like is returning more than 1(please see the next screenshot)

Screenshot 2023-12-13 at 10.54.34.png

Sign up or log in to comment