Create inference.py

inference.py

@@ -0,0 +1,6 @@
+from transformers import pipeline
+
+sentiment_pipeline = pipeline("sentiment-analysis")
+
+def predict(data):
+  sentiment_pipeline(data)