Update app/main.py

model assign issue fixed

app/main.py

@@ -15,7 +15,8 @@ app = FastAPI()
 
 model = BertForSequenceClassification.from_pretrained(
     "JaySenpai/bert-model",
-    cache_dir="/code/cache"
+    cache_dir="/code/cache",
+    use_safetensors=True  
 )
 tokenizer = BertTokenizer.from_pretrained(
     "JaySenpai/bert-model",
@@ -31,7 +32,7 @@ class TextInput(BaseModel):
 
 @app.post("/predict")
 async def predict(data: TextInput):
-    text = data.text  # ✅ Dot notation
+    text = data.text  
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
         outputs = model(**inputs)