Update app.py
Browse files
app.py
CHANGED
|
@@ -24,6 +24,7 @@ class MultiTaskModel(nn.Module):
|
|
| 24 |
return topik_logits, sentimen_logits
|
| 25 |
|
| 26 |
|
|
|
|
| 27 |
# Load tokenizer & model
|
| 28 |
tokenizer = AutoTokenizer.from_pretrained("tokenizer")
|
| 29 |
model = MultiTaskModel("indobenchmark/indobert-base-p1", num_topic_classes=5, num_sentiment_classes=3)
|
|
|
|
| 24 |
return topik_logits, sentimen_logits
|
| 25 |
|
| 26 |
|
| 27 |
+
|
| 28 |
# Load tokenizer & model
|
| 29 |
tokenizer = AutoTokenizer.from_pretrained("tokenizer")
|
| 30 |
model = MultiTaskModel("indobenchmark/indobert-base-p1", num_topic_classes=5, num_sentiment_classes=3)
|