w11wo commited on
Commit
735c4ef
1 Parent(s): c2bc91b

Fix Auth Token

Browse files
Files changed (1) hide show
  1. model.py +8 -8
model.py CHANGED
@@ -21,10 +21,10 @@ models = {
21
  "desc": "A tool to showcase the full capabilities of text analysis NusaBERT fine-tuning has to offer.",
22
  "interface": text_analysis_interface,
23
  "pipe": [
24
- pipeline(model="LazarusNLP/NusaBERT-base-EmoT", use_auth_token=auth_token),
25
- pipeline(model="LazarusNLP/NusaBERT-base-EmoT", use_auth_token=auth_token),
26
- pipeline(model="LazarusNLP/NusaBERT-base-POSP", use_auth_token=auth_token),
27
- pipeline(model="LazarusNLP/NusaBERT-base-NERP", use_auth_token=auth_token),
28
  ],
29
  },
30
  "Sentiment Analysis": {
@@ -37,7 +37,7 @@ models = {
37
  "output_label": "Sentiment Analysis",
38
  "desc": "A sentiment-text-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's SmSA dataset consisting of Indonesian comments and reviews.",
39
  "interface": text_interface,
40
- "pipe": pipeline(model="LazarusNLP/NusaBERT-base-EmoT", use_auth_token=auth_token),
41
  },
42
  "Emotion Detection": {
43
  "title": "Emotion Classifier",
@@ -49,7 +49,7 @@ models = {
49
  "output_label": "Emotion Classifier",
50
  "desc": "An emotion classifier based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's EmoT dataset",
51
  "interface": text_interface,
52
- "pipe": pipeline(model="LazarusNLP/NusaBERT-base-EmoT", use_auth_token=auth_token),
53
  },
54
  "POS Tagging": {
55
  "title": "POS Tagging",
@@ -61,7 +61,7 @@ models = {
61
  "output_label": "POS Tagging",
62
  "desc": "A part-of-speech token-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's POSP dataset consisting of tag-labelled news.",
63
  "interface": token_classification_interface,
64
- "pipe": pipeline(model="LazarusNLP/NusaBERT-base-POSP", use_auth_token=auth_token),
65
  },
66
  "NER Tagging": {
67
  "title": "NER Tagging",
@@ -73,6 +73,6 @@ models = {
73
  "output_label": "NER Tagging",
74
  "desc": "A NER Tagging token-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's NERP dataset consisting of tag-labelled news.",
75
  "interface": token_classification_interface,
76
- "pipe": pipeline(model="LazarusNLP/NusaBERT-base-NERP", use_auth_token=auth_token),
77
  },
78
  }
 
21
  "desc": "A tool to showcase the full capabilities of text analysis NusaBERT fine-tuning has to offer.",
22
  "interface": text_analysis_interface,
23
  "pipe": [
24
+ pipeline(model="LazarusNLP/NusaBERT-base-EmoT", token=auth_token),
25
+ pipeline(model="LazarusNLP/NusaBERT-base-EmoT", token=auth_token),
26
+ pipeline(model="LazarusNLP/NusaBERT-base-POSP", token=auth_token),
27
+ pipeline(model="LazarusNLP/NusaBERT-base-NERP", token=auth_token),
28
  ],
29
  },
30
  "Sentiment Analysis": {
 
37
  "output_label": "Sentiment Analysis",
38
  "desc": "A sentiment-text-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's SmSA dataset consisting of Indonesian comments and reviews.",
39
  "interface": text_interface,
40
+ "pipe": pipeline(model="LazarusNLP/NusaBERT-base-EmoT", token=auth_token),
41
  },
42
  "Emotion Detection": {
43
  "title": "Emotion Classifier",
 
49
  "output_label": "Emotion Classifier",
50
  "desc": "An emotion classifier based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's EmoT dataset",
51
  "interface": text_interface,
52
+ "pipe": pipeline(model="LazarusNLP/NusaBERT-base-EmoT", token=auth_token),
53
  },
54
  "POS Tagging": {
55
  "title": "POS Tagging",
 
61
  "output_label": "POS Tagging",
62
  "desc": "A part-of-speech token-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's POSP dataset consisting of tag-labelled news.",
63
  "interface": token_classification_interface,
64
+ "pipe": pipeline(model="LazarusNLP/NusaBERT-base-POSP", token=auth_token),
65
  },
66
  "NER Tagging": {
67
  "title": "NER Tagging",
 
73
  "output_label": "NER Tagging",
74
  "desc": "A NER Tagging token-classification model based on the BERT model. The model was originally the pre-trained NusaBERT Base model, which is then fine-tuned on indonlu's NERP dataset consisting of tag-labelled news.",
75
  "interface": token_classification_interface,
76
+ "pipe": pipeline(model="LazarusNLP/NusaBERT-base-NERP", token=auth_token),
77
  },
78
  }