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