kadabengaran commited on
Commit
da852e5
·
1 Parent(s): 8ee03b1

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -10,7 +10,7 @@ except Exception as e:
10
  print(e)
11
 
12
  # Config
13
- MODELS_PATH = "kadabengaran/IndoBERT-BiLSTM-Useful-App-Review"
14
 
15
  id2label= {0: 'Other', 1: 'Problem Discovery', 2: 'Information Seeking', 3: 'Feature Request'}
16
  label2id= {'Other': 0, 'Problem Discovery': 1, 'Information Seeking': 2, 'Feature Request': 3}
 
10
  print(e)
11
 
12
  # Config
13
+ MODELS_PATH = "kadabengaran/distilbert-base-uncased-lora-text-classification"
14
 
15
  id2label= {0: 'Other', 1: 'Problem Discovery', 2: 'Information Seeking', 3: 'Feature Request'}
16
  label2id= {'Other': 0, 'Problem Discovery': 1, 'Information Seeking': 2, 'Feature Request': 3}