Spaces:
Running
Running
adamtayzzz
commited on
Commit
•
ecfd465
1
Parent(s):
e8f6a9d
Update app.py
Browse files
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()
|