Spaces:
Runtime error
Runtime error
File size: 304 Bytes
d563a73 |
1 2 3 4 5 6 7 8 |
from transformers import pipeline
# sentiment_pipeline = pipeline("sentiment-analysis")
# data = ["I love you", "I hate you"]
def predict(data, custom_model: str ="finiteautomata/bertweet-base-sentiment-analysis"):
sentiment_pipeline = pipeline("sentiment-analysis")
return sentiment_pipeline(data)
|