Update app.py

app.py

@@ -1,5 +1,6 @@
 import gensim.downloader 
 import gradio as gr
+import pandas as pd
 model = gensim.downloader.load("glove-wiki-gigaword-50")
 
 
@@ -18,11 +19,16 @@ Word1 = gr.Textbox()
 Word2 = gr.Textbox()
 Word3 = gr.Textbox()
 label = gr.Label(show_label=True, label="Word4")
+bp = gr.BarPlot()
 
 def inference(word1, word2, word3):
     output = model.similar_by_vector(model[word3] + model[word2] - model[word1])
-    out_str = [x + ": " + str(y) for x,y in [item for item in output]]
-    return """\n""".join(out_str)
+    #out_str = [x + ": " + str(y) for x,y in [item for item in output]]
+    #return """\n""".join(out_str)
+    df = pd.DataFrame({"x": [x for x,y in [item for item in output],
+                       "y": [str(y) for x,y in [item for item in output]
+                      })
+    return df
 
 examples = [
     ["woman", "man", "aunt"],  
@@ -33,7 +39,7 @@ examples = [
 iface = gr.Interface(
     fn=inference,
     inputs=[Word1, Word2, Word3],
-    outputs=label,
+    outputs=bp,
     description=description,
     examples=examples
     )