pzangara commited on
Commit
2d7d60d
1 Parent(s): 7ce9bf2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,15 +11,15 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
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"]
15
  output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
16
  return output
17
 
18
 
19
  iface = gr.Interface(
20
  fn=clasificador,
21
- inputs=gr.inputs.Textbox(label="Escriba algo en referencia a una ciudad"),
22
- outputs=gr.outputs.Textbox(label="Clasificaci贸n de geografia urbana factible")
23
  )
24
 
25
  iface.launch()
 
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 = ["consumidor","ventas","ganancias","cliente","prestamo","dinero"]
15
  output = classifier(sequence_to_classify, candidate_labels, multi_label=False)
16
  return output
17
 
18
 
19
  iface = gr.Interface(
20
  fn=clasificador,
21
+ inputs=gr.inputs.Textbox(label="Ingrese texto"),
22
+ outputs=gr.outputs.Textbox(label="Clasificaci贸n")
23
  )
24
 
25
  iface.launch()