cpu set

app.py

@@ -15,7 +15,7 @@ CORS(app, resources={r"api/predict/*": {"origins": ["http://localhost:3000", "ht
 # Global variables for model and tokenizer
 model = None
 tokenizer = None
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")  # Use GPU if available
+device = torch.device("cpu")  # Explicitly use CPU
 
 def get_model_and_tokenizer(model_id):
     global model, tokenizer