Update app.py

app.py

@@ -64,7 +64,7 @@ classifier = MyClassifier(model,tokenizer,label_list,output_mode,exit_type,exit_
 
 def greet(text,text2):
     text_input = [(text,text2)]
-    return classifier.get_prob(text_input)
+    return classifier.get_prob(text_input),classifier.get_current_exit()
 
-iface = gr.Interface(fn=greet, inputs=["text","text"], outputs="text")
+iface = gr.Interface(fn=greet, inputs=["text","text"], outputs=["text","number"])
 iface.launch()