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