Update app.py

app.py

@@ -5,10 +5,8 @@ import torch
 
 os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
-# Initialize
 app = Flask(__name__)
 
-# Load model and tokenizer once
 MODEL_NAME = "s-nlp/roberta-base-formality-ranker"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
@@ -40,10 +38,8 @@ def predict_formality():
 
     return jsonify({
         "text": text,
-        "formality_score": round(score, 3),
         "formal_percent": formal_percent,
-        "informal_percent": informal_percent,
-        "classification": classification
+        "informal_percent": informal_percent
     })
 
 if __name__ == "__main__":