Update app.py

app.py

@@ -1,7 +1,20 @@
 import gradio as gr
+from gensim.models import KeyedVectors
 
-def greet(name):
-    return "Hello " + name + "!!"
+def isNoneWords(word):
+    if word is None or len(word)==0 or word not in model.vocab:
+        return True
+    else:
+        return False
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def top_similarity_route(word):
+    if isNoneWords(word):
+        return "word is null or not in model!"
+    else:
+        return {'word':word,'top_similar_words':model.similar_by_word(word, topn=20, restrict_vocab=None)}
+
+
+if __name__ == '__main__':
+    model = KeyedVectors.load_word2vec_format('tencent-ailab-embedding-zh-d100-v0.2.0-s.txt', binary=False)
+    iface = gr.Interface(fn=top_similarity_route, inputs="text", outputs="text")
+    iface.launch()