Update app.py

app.py

@@ -42,13 +42,13 @@ except Exception as e:
     print("⚠ Could not load config.json — using fallback")
     id2label = {
         0: "Politics",
-        1: "Business",
+        1: "Economy",
         2: "Sports",
         3: "Technology",
         4: "Health",
-        5: "Entertainment",
-        6: "World",
-        7: "Other",
+        5: "Agriculture",
+        6: "accident",
+        7: "education",
     }
     label2id = {v: k for k, v in id2label.items()}
     num_labels = len(id2label)
@@ -111,10 +111,10 @@ tokenizer = AutoTokenizer.from_pretrained(PRETRAINED)
 
 model = HybridSentenceChuLo(PRETRAINED, num_labels).to(DEVICE)
 
-print("Loading trained weights...")
-state = torch.hub.load_state_dict_from_url(
-    f"https://huggingface.co/{HF_MODEL_ID}/resolve/main/pytorch_model.bin",
-    map_location=DEVICE
+from transformers import AutoModel
+model = AutoModel.from_pretrained(
+    "Abelex/afro-xlmr-large",
+    trust_remote_code=True
 )
 model.load_state_dict(state, strict=False)
 model.eval()