Update app.py

app.py

@@ -10,7 +10,7 @@ class MultiTaskModel(nn.Module):
         self.encoder = AutoModel.from_pretrained(base_model_name)
         hidden_size = self.encoder.config.hidden_size
         self.topik_classifier = nn.Linear(hidden_size, num_topic_classes)
-        self.sentimen_classifier = nn.Linear(hidden_size, num_sentiment_classes)
+        self.sentiment_classifier = nn.Linear(hidden_size, num_sentiment_classes)
 
     def forward(self, input_ids, attention_mask):
         outputs = self.encoder(input_ids=input_ids, attention_mask=attention_mask)
@@ -21,7 +21,7 @@ class MultiTaskModel(nn.Module):
 
 # Load tokenizer & model
 tokenizer = AutoTokenizer.from_pretrained("tokenizer")
-model = MultiTaskModel("indobenchmark/indobert-base-p1", num_topic_classes=4, num_sentiment_classes=3)
+model = MultiTaskModel("indobenchmark/indobert-base-p1", num_topic_classes=5, num_sentiment_classes=3)
 model.load_state_dict(torch.load("model.pt", map_location=torch.device("cpu")))
 model.eval()