kadabengaran commited on
Commit
af9f658
·
1 Parent(s): b767f00
Files changed (1) hide show
  1. 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
- # 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
- # "kadabengaran/IndoBERT-BiLSTM-Useful-App-Review"]
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