pzangara commited on
Commit
16ec01c
1 Parent(s): 7d036b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,10 +5,10 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
5
  import transformers
6
  import gradio as gr
7
 
8
- model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
9
- tokenizer = AutoTokenizer.from_pretrained(model_name)
10
 
11
  def clasificador(input):
 
 
12
  classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli",tokenizer=tokenizer)
13
  sequence_to_classify = input
14
  candidate_labels = ["Nueva York","Barcelona","Buenos Aires","C贸rdoba","Roma","Londres","Sidney"]
 
5
  import transformers
6
  import gradio as gr
7
 
 
 
8
 
9
  def clasificador(input):
10
+ model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
11
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
12
  classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v3-base-mnli-xnli",tokenizer=tokenizer)
13
  sequence_to_classify = input
14
  candidate_labels = ["Nueva York","Barcelona","Buenos Aires","C贸rdoba","Roma","Londres","Sidney"]