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() |