import time | |
from transformers import pipeline | |
pipeline = pipeline("sentiment-analysis", model="juliensimon/autonlp-imdb-demo-hf-16622767", use_auth_token=True) | |
data = "Dune is something rarely seen anymore – it’s an epic." | |
tick=time.time_ns() | |
for i in range(0,100): | |
response=pipeline(data) | |
tock=time.time_ns() | |
print("%.2fms" % ((tock-tick)/1000/1000/100)) | |