File size: 405 Bytes
635e914
 
 
 
 
 
 
31d6241
635e914
 
 
1
2
3
4
5
6
7
8
9
10
11
12
import gradio as gr
import pandas as pd
import gensim.downloader as api
wv = api.load('glove-wiki-gigaword-50')

    
def get_associations(term,top_words=10):
    return pd.DataFrame(wv.most_similar(positive=[term], topn=10),columns=['word','similarity'])


iface = gr.Interface(fn=get_associations, inputs="text", outputs=gr.Dataframe(headers=['word','similarity']),examples=["cat"]).launch(share=False)