Update app/main.py

app/main.py

@@ -1,7 +1,7 @@
 import os
 from fastapi import FastAPI
 from pydantic import BaseModel
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from transformers import BertTokenizer, BertForSequenceClassification
 from sklearn.preprocessing import LabelEncoder
 import torch
 import numpy as np
@@ -11,14 +11,12 @@ os.environ["TRANSFORMERS_CACHE"] = "/code/cache"
 
 app = FastAPI()
 
-model = AutoModelForSequenceClassification.from_pretrained(
-    "JaySenpai/bert-model",
-    cache_dir="/code/cache",
-    use_safetensors=True
+model = BertForSequenceClassification.from_pretrained(
+    "./bert-model"  # or adjust path based on your structure
 )
-tokenizer = AutoTokenizer.from_pretrained(
-    "JaySenpai/bert-model",
-    cache_dir="/code/cache"
+
+tokenizer = BertTokenizer.from_pretrained(
+    "./bert-model"
 )
 model.eval()