fixing

Procfile

@@ -1 +1 @@
-web: gunicorn -b 0.0.0.0:8001 app:app
+web: gunicorn app:app

api/predict.py

@@ -1,20 +0,0 @@
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
-import torch
-
-model_name = "w11wo/indonesian-roberta-base-sentiment-classifier"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSequenceClassification.from_pretrained(model_name)
-
-labels = ["positive", "negative", "neutral"]
-
-def softmax(logits):
-    exp_logits = torch.exp(logits)
-    return exp_logits / torch.sum(exp_logits, dim=1, keepdim=True)
-
-def predict(text):
-    inputs = tokenizer(text, return_tensors="pt")
-    outputs = model(**inputs)
-    probabilities = softmax(outputs.logits)
-    probs = probabilities.detach().numpy().tolist()[0]
-    label = labels[probs.index(max(probs))]
-    return label

app.py

@@ -1,14 +1,30 @@
 from flask import Flask, request, jsonify
-from api.predict import predict
+from transformers import pipeline
 
+# Inisialisasi aplikasi Flask
 app = Flask(__name__)
 
-@app.route('/predict', methods=['POST'])
-def predict_route():
-    data = request.get_json()
-    text = data.get('text')
-    result = predict(text)
-    return jsonify({"sentiment": result})
+# Inisialisasi pipeline untuk sentiment analysis
+sentiment_analysis = pipeline("sentiment-analysis", model="w11wo/indonesian-roberta-base-sentiment-classifier")
+
+@app.route('/')
+def home():
+    return 'API Sentiment Analysis menggunakan w11wo/indonesian-roberta-base-sentiment-classifier'
+
+@app.route('/analyze', methods=['POST'])
+def analyze_sentiment():
+    try:
+        data = request.json
+        if 'text' not in data:
+            return jsonify({"error": "No text provided"}), 400
+
+        text = data['text']
+        result = sentiment_analysis(text)
+        
+        return jsonify(result), 200
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=8001)
+    app.run(host='0.0.0.0', port=8080)

requirements.txt

@@ -1,4 +1,4 @@
 flask
 transformers
 torch
-gunicorn
+gunicorn