word2vec / app.py
Lyem's picture
Update app.py
745ed38
import gensim
import gradio as gr
model = gensim.models.Word2Vec.load("w2v_model.model")
def most_sim(input):
sim_lst = model.wv.most_similar(input)
sim_str = ""
for tup in sim_lst:
sim_str = sim_str + str(tup) + "\n"
return sim_str
demo = gr.Interface(fn=most_sim, inputs="text", outputs="text")
demo.launch()