Update app.py

app.py

@@ -17,9 +17,9 @@ def test_input(words):
     word_dict = {}
     for w in words.split(","):
         if w in all_set:
-            word_dict[w] = 1.
+            word_dict[w] = "in-distribution"
         else:
-            word_dict[w] = 0.
+            word_dict[w] = "out-distribution"
     return word_dict
 
     
@@ -58,7 +58,7 @@ examples_w = [["red,car,vehicle"], ["fake,smile"], ["small,cat,animal"]]
 interface_words = gr.Interface(
             fn=test_input,
             inputs=gr.Textbox(label="Input:word_1,word2,...,word_n"),
-            outputs=gr.outputs.Label(label="In training-distribution?"),
+            outputs=gr.Textbox(label="In training-distribution?"),
             description=description_w,
             examples=examples_w,
 )