Spaces:

SuperBigtoo
/

ThaiNewsClassify

Sleeping

SuperBigtoo commited on Oct 21, 2023

Commit

1f99b04

1 Parent(s): 225444a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

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