Spaces:

SuperBigtoo
/

ThaiNewsClassify

Sleeping

SuperBigtoo commited on Oct 21, 2023

Commit

e681492

1 Parent(s): 6338950

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

 loaded_model = ClassificationModel(
      "bert",
+     "/model/thainewsClassify_model_3_14",
      use_cuda=torch.cuda.is_available(),
      num_labels=14,
 )