Commit
·
296d4cd
1
Parent(s):
3227970
Update app.py
Browse fileschange the model into a multilingual model
app.py
CHANGED
@@ -10,7 +10,7 @@ class TaskClassifier:
|
|
10 |
|
11 |
def __init__(self):
|
12 |
self.classifier = pipeline("zero-shot-classification",
|
13 |
-
model="
|
14 |
|
15 |
def __call__(self, client_input: str, task_types: str):
|
16 |
"""Classify tasks for LLM-based gent"""
|
|
|
10 |
|
11 |
def __init__(self):
|
12 |
self.classifier = pipeline("zero-shot-classification",
|
13 |
+
model="MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7")
|
14 |
|
15 |
def __call__(self, client_input: str, task_types: str):
|
16 |
"""Classify tasks for LLM-based gent"""
|