adamtayzzz commited on
Commit
ecfd465
1 Parent(s): e8f6a9d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ classifier = MyClassifier(model,tokenizer,label_list,output_mode,exit_type,exit_
64
 
65
  def greet(text,text2):
66
  text_input = [(text,text2)]
67
- return classifier.get_prob(text_input)
68
 
69
- iface = gr.Interface(fn=greet, inputs=["text","text"], outputs="text")
70
  iface.launch()
 
64
 
65
  def greet(text,text2):
66
  text_input = [(text,text2)]
67
+ return classifier.get_prob(text_input),classifier.get_current_exit()
68
 
69
+ iface = gr.Interface(fn=greet, inputs=["text","text"], outputs=["text","number"])
70
  iface.launch()