Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
-
-os.environ["TRANSFORMERS_CACHE"] = "/tmp"
-
 from flask import Flask, request, jsonify
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
+# Fix cache permission issue
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+
 app = Flask(__name__)
 
 # Load the model and tokenizer
@@ -13,6 +13,11 @@ model_name = "s-nlp/roberta-base-formality-ranker"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
 
+# Route for testing the API
+@app.route("/", methods=["GET"])
+def home():
+    return jsonify({"message": "Formality Classifier API is running! Use /predict to classify text."})
+
 # Fuzzy classification function
 def fuzzy_formality(score, threshold=0.75):
     if score < threshold: