Spaces:
Runtime error
Runtime error
Fix Auth Token
Browse files
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",
|
25 |
-
pipeline(model="LazarusNLP/NusaBERT-base-EmoT",
|
26 |
-
pipeline(model="LazarusNLP/NusaBERT-base-POSP",
|
27 |
-
pipeline(model="LazarusNLP/NusaBERT-base-NERP",
|
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",
|
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",
|
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",
|
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",
|
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 |
}
|