pzangara commited on
Commit
7a59383
1 Parent(s): 69e3f68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -5,6 +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
 
@@ -12,4 +16,6 @@ classifier = pipeline("zero-shot-classification", model="MoritzLaurer/mDeBERTa-v
12
  sequence_to_classify = "Meti贸 5 goles en dos minutos"
13
  candidate_labels = ["futbol","rugby","basket","tenis","golf","automovilismo","ciclismo"]
14
  output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
15
- print(output)
 
 
 
5
  import transformers
6
  import gradio as gr
7
 
8
+
9
+ def image_classifier(inp):
10
+ return {'cat': 0.3, 'dog': 0.7}
11
+
12
  model_name = "MoritzLaurer/mDeBERTa-v3-base-xnli-multilingual-nli-2mil7"
13
  tokenizer = AutoTokenizer.from_pretrained(model_name)
14
 
 
16
  sequence_to_classify = "Meti贸 5 goles en dos minutos"
17
  candidate_labels = ["futbol","rugby","basket","tenis","golf","automovilismo","ciclismo"]
18
  output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
19
+
20
+ demo = gr.Interface(fn=classifier, inputs=(sequence_to_classify, candidate_labels, multi_label=False), outputs="label")
21
+ demo.launch()