Update app.py

app.py

@@ -31,7 +31,7 @@ class IndoBERTClassifier(nn.Module):
 
 # ===== Load Model and Tokenizer =====
 tokenizer = AutoTokenizer.from_pretrained("indobenchmark/indobert-base-p1")
-model = IndoBERTClassifier("indobenchmark/indobert-base-p1", num_labels=4)
+model = IndoBERTClassifier("indobenchmark/indobert-base-p1", num_labels=5)
 model.load_state_dict(torch.load("pytorch_model.bin", map_location=torch.device("cpu")))
 model.eval()