Update app.py

app.py

@@ -1,45 +1,32 @@
-from flask import Flask, request, jsonify
+import os
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from flask import Flask, request, jsonify
+
+# Set cache dir ke folder yang bisa ditulis
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
-# Load model sekali aja
 MODEL_NAME = "taufiqdp/indonesian-sentiment"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+
+# Load model dan tokenizer
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir="/tmp")
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, cache_dir="/tmp")
 
 class_names = ['negatif', 'netral', 'positif']
 
 app = Flask(__name__)
 
-@app.route("/predict", methods=["POST"])
+@app.route("/", methods=["POST"])
 def predict():
-    try:
-        data = request.get_json()
-        text = data.get("text", "")
-
-        if not text.strip():
-            return jsonify({"error": "Teks kosong"}), 400
-
-        tokenized = tokenizer(text, return_tensors="pt")
-        with torch.inference_mode():
-            logits = model(**tokenized).logits
-
-        pred_id = logits.argmax(dim=1).item()
-        sentiment = class_names[pred_id]
-        confidence = torch.softmax(logits, dim=1)[0][pred_id].item()
-
-        return jsonify({
-            "sentiment": sentiment,
-            "confidence": round(confidence, 4)
-        })
-    except Exception as e:
-        return jsonify({"error": str(e)}), 500
-
+    data = request.get_json()
+    text = data.get("inputs", "")
 
-@app.route("/", methods=["GET"])
-def home():
-    return jsonify({"message": "Indonesian Sentiment API ready!"})
+    tokenized_text = tokenizer(text, return_tensors='pt')
+    with torch.inference_mode():
+        logits = model(**tokenized_text).logits
 
+    result = class_names[logits.argmax(dim=1).item()]
+    return jsonify([{"label": result, "score": float(torch.softmax(logits, dim=1).max().item())}])
 
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)