SinhNguyen commited on
Commit
296d4cd
·
1 Parent(s): 3227970

Update app.py

Browse files

change the model into a multilingual model

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ class TaskClassifier:
10
 
11
  def __init__(self):
12
  self.classifier = pipeline("zero-shot-classification",
13
- model="facebook/bart-large-mnli")
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"""