Update app.py

app.py

@@ -20,8 +20,8 @@ print(f"Using device: {device}")
 print(f"Loading model: {MODEL_ID}...")
 try:
     # Load tokenizer and model, providing the token for private access
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
-    model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=HF_TOKEN)
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN, trust_remote_code=True)
+    model = AutoModelForSequenceClassification.from_pretrained(MODEL_ID, token=HF_TOKEN,trust_remote_code=True, num_labels=3)
     
     # Move the model to the selected device ONCE for efficiency
     model.to(device)