Update app.py

app.py

@@ -1,11 +1,11 @@
-from flask import Flask, request, jsonify
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
-import torch
 import os
 
-# Set writable cache
 os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 
+from flask import Flask, request, jsonify
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import torch
+
 app = Flask(__name__)
 
 # Load the model and tokenizer
@@ -58,6 +58,6 @@ def predict_formality():
         **result
     })
 
-# Run on correct port
+# Ensure Flask runs on the correct port
 if __name__ == "__main__":
     app.run(host="0.0.0.0", port=7860)