10zinten commited on
Commit
21b168b
1 Parent(s): 081ddd8

fix version

Browse files
Files changed (2) hide show
  1. app.py +27 -25
  2. requirements.txt +1 -2
app.py CHANGED
@@ -23,34 +23,36 @@ def find_most_similar(word):
23
  sim_output = wv.most_similar(word)
24
  return format_to_html(sim_output)
25
 
26
- def similarity(wordA, wordB):
27
- return wv.similarity(wordA, wordB)
28
 
29
  title = "Classical Bo Word2Vec word vectors"
30
  examples = ["སེམས་", "བླ་མ་", "སྟོབས་"]
31
 
32
- # demo = gr.Interface(
33
- # fn=find_most_similar,
34
- # inputs="text",
35
- # outputs="html",
36
- # title=title,
37
- # examples=examples
38
- # ).launch()
39
-
40
- demo = gr.Blocks()
41
-
42
- with demo:
43
- # Find top 10 most similar words
44
- print(gr.__version__)
45
- word = gr.inputs.Textbox("Word")
46
- top_10_similar_words = gr.HTML()
47
- b1 = gr.Button("Find similar words")
48
- b1.click(find_most_similar, inputs=word, outputs=top_10_similar_words)
 
 
49
 
50
- wordA = gr.inputs.Textbox("Word A")
51
- wordB = gr.inputs.Textbox("Word B")
52
- similarity = gr.Number()
53
- b2 = gr.Button("Find similarity")
54
- b2.click(similarity, inputs=[wordA, wordB], outputs=similarity)
55
 
56
- demo.launch()
23
  sim_output = wv.most_similar(word)
24
  return format_to_html(sim_output)
25
 
26
+ def find_similarity(wordA, wordB):
27
+ return max([wv.similarity(wordA, wordB), 0])
28
 
29
  title = "Classical Bo Word2Vec word vectors"
30
  examples = ["སེམས་", "བླ་མ་", "སྟོབས་"]
31
 
32
+ print(find_similarity(examples[0], examples[1]))
33
+
34
+ demo = gr.Interface(
35
+ fn=find_similarity,
36
+ inputs=[gr.inputs.Textbox("Word A"), gr.inputs.Textbox("Word B")],
37
+ outputs="number",
38
+ title=title,
39
+ # examples=examples
40
+ ).launch()
41
+
42
+ # demo = gr.Blocks()
43
+
44
+ # with demo:
45
+ # # Find top 10 most similar words
46
+ # print(gr.__version__)
47
+ # word = gr.inputs.Textbox("Word")
48
+ # top_10_similar_words = gr.inputs.HTML()
49
+ # b1 = gr.Button("Find similar words")
50
+ # b1.click(find_most_similar, inputs=word, outputs=top_10_similar_words)
51
 
52
+ # wordA = gr.inputs.Textbox("Word A")
53
+ # wordB = gr.inputs.Textbox("Word B")
54
+ # similarity = gr.Number()
55
+ # b2 = gr.Button("Find similarity")
56
+ # b2.click(similarity, inputs=[wordA, wordB], outputs=similarity)
57
 
58
+ # demo.launch()
requirements.txt CHANGED
@@ -1,2 +1 @@
1
- gensim
2
- gradio==3.0.5
1
+ gensim