filipeclduarte commited on
Commit
dd477af
·
1 Parent(s): effa5d8

Create inference

Browse files
Files changed (1) hide show
  1. inference +6 -0
inference ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ sentiment_pipeline = pipeline("sentiment-analysis")
4
+
5
+ def predict(data):
6
+ sentiment_pipeline(data)