Upload app.py

app.py

@@ -0,0 +1,28 @@
+from flask import Flask, request, jsonify
+import pandas as pd
+from transformers import pipeline
+import io
+
+app = Flask(__name__)
+model = pipeline("sentiment-analysis")
+
+@app.route("/predict", methods=["POST"])
+def predict():
+    data = request.json
+    text = data.get("text")
+    if not text:
+        return jsonify({"error": "No text provided"}), 400
+    result = model(text)
+    return jsonify(result)
+
+@app.route("/predict_csv", methods=["POST"])
+def predict_csv():
+    file = request.files.get("file")
+    if not file:
+        return jsonify({"error": "No CSV uploaded"}), 400
+    df = pd.read_csv(file)
+    df["prediction"] = df["text"].apply(lambda x: model(x)[0]['label'])
+    return df.to_json(orient="records")
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)