minimalFlaskAPI / inference
filipeclduarte's picture
Create inference
dd477af
raw
history blame
133 Bytes
from transformers import pipeline
sentiment_pipeline = pipeline("sentiment-analysis")
def predict(data):
sentiment_pipeline(data)