Spaces:
Runtime error
Runtime error
Commit
·
af9f658
1
Parent(s):
b767f00
fix
Browse files- app/main.py +7 -7
app/main.py
CHANGED
|
@@ -19,16 +19,16 @@ img {
|
|
| 19 |
"""
|
| 20 |
# Config
|
| 21 |
MAX_SEQ_LEN = 128
|
| 22 |
-
bert_path = './local/base-indobert'
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
MODELS_PATH = ["./local/indobert1",
|
| 27 |
-
|
| 28 |
|
| 29 |
MODELS_NAME = ["IndoBERT-BiLSTM", "IndoBERT"]
|
| 30 |
LABELS = {'Not Useful': 0, 'Useful': 1}
|
| 31 |
-
|
| 32 |
HIDDEN_DIM = 768
|
| 33 |
OUTPUT_DIM = 2 # 2 if Binary
|
| 34 |
N_LAYERS = 2
|
|
|
|
| 19 |
"""
|
| 20 |
# Config
|
| 21 |
MAX_SEQ_LEN = 128
|
| 22 |
+
# bert_path = './local/base-indobert'
|
| 23 |
+
bert_path = 'indolem/indobert-base-uncased'
|
| 24 |
+
MODELS_PATH = ["kadabengaran/IndoBERT-Useful-App-Review",
|
| 25 |
+
"kadabengaran/IndoBERT-BiLSTM-Useful-App-Review"]
|
| 26 |
+
# MODELS_PATH = ["./local/indobert1",
|
| 27 |
+
# "./local/indobert2"]
|
| 28 |
|
| 29 |
MODELS_NAME = ["IndoBERT-BiLSTM", "IndoBERT"]
|
| 30 |
LABELS = {'Not Useful': 0, 'Useful': 1}
|
| 31 |
+
|
| 32 |
HIDDEN_DIM = 768
|
| 33 |
OUTPUT_DIM = 2 # 2 if Binary
|
| 34 |
N_LAYERS = 2
|