Spaces:

SuperBigtoo
/

ThaiNewsClassify

Running

SuperBigtoo commited on Oct 24, 2023

Commit

16a516e

•

1 Parent(s): 8add7b5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ typeId = {'การเมือง': 0, 'กีฬา': 1, 'คุณภาพ
 loaded_model = ClassificationModel(
      "bert",
-     "SuperBigtoo/thainews-classificaton-thaikeras-bert-th",
      use_cuda=torch.cuda.is_available(),
      num_labels=14,
 )

 loaded_model = ClassificationModel(
      "bert",
+     "SuperBigtoo/thainews-classification-thaikeras-bert-th",
      use_cuda=torch.cuda.is_available(),
      num_labels=14,
 )