Spaces:

alexandrainst
/

zero-shot-classification

Running

saattrupdan commited on Apr 25

Commit

858b3b9

verified ·

1 Parent(s): 98a2fb0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ def main():
     # Load the zero-shot classification pipeline
     global classifier, model, tokenizer
-    model_id = "alexandrainst/scandi-nli-large"
     model = AutoModelForSequenceClassification.from_pretrained(model_id)
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model.eval()

     # Load the zero-shot classification pipeline
     global classifier, model, tokenizer
+    model_id = "alexandrainst/scandi-nli-large-v2"
     model = AutoModelForSequenceClassification.from_pretrained(model_id)
     tokenizer = AutoTokenizer.from_pretrained(model_id)
     model.eval()