Update app.py

app.py

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