Update app.py

app.py

@@ -1,10 +1,14 @@
+import os
+
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+
 from flask import Flask, request, jsonify
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 
 app = Flask(__name__)
 
-# Load the pretrained model and tokenizer
+# Load the model and tokenizer
 model_name = "s-nlp/roberta-base-formality-ranker"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name)