SuperBigtoo commited on
Commit
1f99b04
1 Parent(s): 225444a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ typeId = {'การเมือง': 0, 'กีฬา': 1, 'คุณภาพ
12
 
13
  loaded_model = ClassificationModel(
14
  "bert",
15
- "/model/thainewsClassify_model_3_14",
16
  use_cuda=torch.cuda.is_available(),
17
  num_labels=14,
18
  )
 
12
 
13
  loaded_model = ClassificationModel(
14
  "bert",
15
+ "SuperBigtoo/thainews-classify",
16
  use_cuda=torch.cuda.is_available(),
17
  num_labels=14,
18
  )