File size: 337 Bytes
8d368c2 2eeb0d8 8d368c2 2eeb0d8 745ed38 2eeb0d8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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() |