Commit
·
d57f7bc
1
Parent(s):
41bb40c
feat: Reduce the amount of candidate_labels for the topic classification
Browse files
app.py
CHANGED
|
@@ -60,45 +60,33 @@ def classification(task: str, doc: str) -> str:
|
|
| 60 |
hypothesis_template = "Detta exempel handlar om {}."
|
| 61 |
candidate_labels = [
|
| 62 |
"krig",
|
| 63 |
-
"regering",
|
| 64 |
"politik",
|
| 65 |
"utbildning",
|
| 66 |
"hälsa",
|
| 67 |
-
"miljö",
|
| 68 |
"ekonomi",
|
| 69 |
-
"affärer",
|
| 70 |
"mode",
|
| 71 |
-
"underhållning",
|
| 72 |
"sport",
|
| 73 |
]
|
| 74 |
elif language == "no":
|
| 75 |
hypothesis_template = "Dette eksemplet handler om {}."
|
| 76 |
candidate_labels = [
|
| 77 |
"krig",
|
| 78 |
-
"myndighetene",
|
| 79 |
"politikk",
|
| 80 |
"utdanning",
|
| 81 |
"helse",
|
| 82 |
-
"miljø",
|
| 83 |
"økonomi",
|
| 84 |
-
"virksomhet",
|
| 85 |
"mote",
|
| 86 |
-
"underholdning",
|
| 87 |
"sport",
|
| 88 |
]
|
| 89 |
else:
|
| 90 |
hypothesis_template = "Denne nyhedsartikel handler primært om {}."
|
| 91 |
candidate_labels = [
|
| 92 |
"krig",
|
| 93 |
-
"regering",
|
| 94 |
"politik",
|
| 95 |
"uddannelse",
|
| 96 |
"sundhed",
|
| 97 |
-
"miljø",
|
| 98 |
"økonomi",
|
| 99 |
-
"forretning",
|
| 100 |
"mode",
|
| 101 |
-
"underholdning",
|
| 102 |
"sport",
|
| 103 |
]
|
| 104 |
|
|
|
|
| 60 |
hypothesis_template = "Detta exempel handlar om {}."
|
| 61 |
candidate_labels = [
|
| 62 |
"krig",
|
|
|
|
| 63 |
"politik",
|
| 64 |
"utbildning",
|
| 65 |
"hälsa",
|
|
|
|
| 66 |
"ekonomi",
|
|
|
|
| 67 |
"mode",
|
|
|
|
| 68 |
"sport",
|
| 69 |
]
|
| 70 |
elif language == "no":
|
| 71 |
hypothesis_template = "Dette eksemplet handler om {}."
|
| 72 |
candidate_labels = [
|
| 73 |
"krig",
|
|
|
|
| 74 |
"politikk",
|
| 75 |
"utdanning",
|
| 76 |
"helse",
|
|
|
|
| 77 |
"økonomi",
|
|
|
|
| 78 |
"mote",
|
|
|
|
| 79 |
"sport",
|
| 80 |
]
|
| 81 |
else:
|
| 82 |
hypothesis_template = "Denne nyhedsartikel handler primært om {}."
|
| 83 |
candidate_labels = [
|
| 84 |
"krig",
|
|
|
|
| 85 |
"politik",
|
| 86 |
"uddannelse",
|
| 87 |
"sundhed",
|
|
|
|
| 88 |
"økonomi",
|
|
|
|
| 89 |
"mode",
|
|
|
|
| 90 |
"sport",
|
| 91 |
]
|
| 92 |
|