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
|