ashishraics commited on
Commit
9111b95
1 Parent(s): b47aba9

change ZS model

Browse files
Files changed (1) hide show
  1. zeroshot_clf_helper.py +1 -1
zeroshot_clf_helper.py CHANGED
@@ -46,7 +46,7 @@ def create_onnx_model_zs(art_path='zeroshot_onnx_dir'):
46
  if not os.path.exists(art_path):
47
  try:
48
  subprocess.run(['python3', '-m', 'transformers.onnx',
49
- '--model=facebook/bart-large-mnli',
50
  '--feature=sequence-classification',
51
  art_path])
52
  except:
 
46
  if not os.path.exists(art_path):
47
  try:
48
  subprocess.run(['python3', '-m', 'transformers.onnx',
49
+ '--model=valhalla/distilbart-mnli-12-1',
50
  '--feature=sequence-classification',
51
  art_path])
52
  except: