Update main.py

main.py

@@ -7,9 +7,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import re
 
 # -----------------------------
-# Load model from local folder
+# Load model from current folder
 # -----------------------------
-MODEL_PATH = "final_shirk_classifier"
+MODEL_PATH = "."  # we are in the repo root
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 # ---------- Text normalization ----------
@@ -33,7 +33,7 @@ label_list = [id2label[i] for i in range(len(id2label))]
 
 SHIRK_LABEL = "shirk"
 SHIRK_INDEX = label_list.index(SHIRK_LABEL)
-SHIRK_THRESHOLD = 0.7
+SHIRK_THRESHOLD = 0.7  # tweak if needed
 
 # ---------- FastAPI ----------
 app = FastAPI(title="Bangla Shirk Classifier API")
@@ -67,7 +67,7 @@ def predict(req: PredictRequest):
         logits = outputs.logits[0]
         probs = F.softmax(logits, dim=-1).cpu().numpy()
 
-    # apply threshold logic
+    # Shirk threshold logic
     top1 = int(probs.argmax())
     if top1 == SHIRK_INDEX and probs[SHIRK_INDEX] < SHIRK_THRESHOLD:
         top2 = int(probs.argsort()[-2])