Spaces:
Runtime error
Runtime error
Commit
·
c22dcfd
1
Parent(s):
203952e
Upload 8 files
Browse files- .gitattributes +1 -4
- README.md +4 -4
- app.py +29 -0
- inference_dict_for_word2vec.pkl +3 -0
- requirements.txt +0 -0
- text_to_seq.pkl +3 -0
- web_pages/index.html +16 -0
- word2vec_65_epochs.h5 +3 -0
.gitattributes
CHANGED
@@ -2,13 +2,11 @@
|
|
2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
5 |
-
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
6 |
*.ftz filter=lfs diff=lfs merge=lfs -text
|
7 |
*.gz filter=lfs diff=lfs merge=lfs -text
|
8 |
*.h5 filter=lfs diff=lfs merge=lfs -text
|
9 |
*.joblib filter=lfs diff=lfs merge=lfs -text
|
10 |
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
11 |
-
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
12 |
*.model filter=lfs diff=lfs merge=lfs -text
|
13 |
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
14 |
*.npy filter=lfs diff=lfs merge=lfs -text
|
@@ -16,13 +14,12 @@
|
|
16 |
*.onnx filter=lfs diff=lfs merge=lfs -text
|
17 |
*.ot filter=lfs diff=lfs merge=lfs -text
|
18 |
*.parquet filter=lfs diff=lfs merge=lfs -text
|
19 |
-
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
*.pickle filter=lfs diff=lfs merge=lfs -text
|
21 |
*.pkl filter=lfs diff=lfs merge=lfs -text
|
|
|
22 |
*.pt filter=lfs diff=lfs merge=lfs -text
|
23 |
*.pth filter=lfs diff=lfs merge=lfs -text
|
24 |
*.rar filter=lfs diff=lfs merge=lfs -text
|
25 |
-
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
26 |
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
27 |
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
28 |
*.tflite filter=lfs diff=lfs merge=lfs -text
|
|
|
2 |
*.arrow filter=lfs diff=lfs merge=lfs -text
|
3 |
*.bin filter=lfs diff=lfs merge=lfs -text
|
4 |
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
|
|
5 |
*.ftz filter=lfs diff=lfs merge=lfs -text
|
6 |
*.gz filter=lfs diff=lfs merge=lfs -text
|
7 |
*.h5 filter=lfs diff=lfs merge=lfs -text
|
8 |
*.joblib filter=lfs diff=lfs merge=lfs -text
|
9 |
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
|
|
10 |
*.model filter=lfs diff=lfs merge=lfs -text
|
11 |
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
12 |
*.npy filter=lfs diff=lfs merge=lfs -text
|
|
|
14 |
*.onnx filter=lfs diff=lfs merge=lfs -text
|
15 |
*.ot filter=lfs diff=lfs merge=lfs -text
|
16 |
*.parquet filter=lfs diff=lfs merge=lfs -text
|
|
|
17 |
*.pickle filter=lfs diff=lfs merge=lfs -text
|
18 |
*.pkl filter=lfs diff=lfs merge=lfs -text
|
19 |
+
*.pb filter=lfs diff=lfs merge=lfs -text
|
20 |
*.pt filter=lfs diff=lfs merge=lfs -text
|
21 |
*.pth filter=lfs diff=lfs merge=lfs -text
|
22 |
*.rar filter=lfs diff=lfs merge=lfs -text
|
|
|
23 |
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
24 |
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
25 |
*.tflite filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
@@ -1,10 +1,10 @@
|
|
1 |
---
|
2 |
title: Similarity Word2vec
|
3 |
-
emoji:
|
4 |
-
colorFrom:
|
5 |
-
colorTo:
|
6 |
sdk: gradio
|
7 |
-
sdk_version: 3.
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
|
|
1 |
---
|
2 |
title: Similarity Word2vec
|
3 |
+
emoji: 🐠
|
4 |
+
colorFrom: indigo
|
5 |
+
colorTo: purple
|
6 |
sdk: gradio
|
7 |
+
sdk_version: 3.1.4
|
8 |
app_file: app.py
|
9 |
pinned: false
|
10 |
---
|
app.py
ADDED
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
|
2 |
+
|
3 |
+
import pickle
|
4 |
+
from tensorflow.keras.models import load_model
|
5 |
+
from scipy import spatial
|
6 |
+
import gradio as gr
|
7 |
+
|
8 |
+
model = load_model('word2vec_65_epochs.h5')
|
9 |
+
model.pop()
|
10 |
+
with open('text_to_seq.pkl', 'rb') as f:
|
11 |
+
tokenizer = pickle.load(f)
|
12 |
+
with open('inference_dict_for_word2vec.pkl', 'rb') as f:
|
13 |
+
inference_dict = pickle.load(f)
|
14 |
+
|
15 |
+
|
16 |
+
def get_answer(text):
|
17 |
+
similarities = []
|
18 |
+
ans = []
|
19 |
+
seq = tokenizer.texts_to_sequences([text])
|
20 |
+
pred = model.predict(seq)
|
21 |
+
for j,i in enumerate(model.layers[0].get_weights()[0]):
|
22 |
+
similarities.append([(1 - spatial.distance.cosine(pred[0], i)),j])
|
23 |
+
similarities = sorted(similarities,key=lambda x: x[0],reverse=True)
|
24 |
+
for i in range(10):
|
25 |
+
ans.append((inference_dict[similarities[i][1]],similarities[i][0]))
|
26 |
+
return f"{ans}"
|
27 |
+
|
28 |
+
demo = gr.Interface(fn=get_answer, inputs="text", outputs="label")
|
29 |
+
demo.launch()
|
inference_dict_for_word2vec.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9c530aefe2c3699bfa1f39064dbc543a8bb9203b215eabb200c7cae16d7c7854
|
3 |
+
size 1829262
|
requirements.txt
ADDED
Binary file (6.44 kB). View file
|
|
text_to_seq.pkl
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1a21bb2620d33fc65fa716fda5246d54d2626de00248056b6b5530d9769e9ff6
|
3 |
+
size 5296416
|
web_pages/index.html
ADDED
@@ -0,0 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
<!DOCTYPE html>
|
2 |
+
<html lang="en">
|
3 |
+
<head>
|
4 |
+
<meta charset="UTF-8">
|
5 |
+
<meta http-equiv="X-UA-Compatible" content="IE=edge">
|
6 |
+
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
7 |
+
<title>Find Similar Words</title>
|
8 |
+
</head>
|
9 |
+
<body>
|
10 |
+
<form action="/predict" method="post">
|
11 |
+
Enter a word: <input name="input_data" type="text"/>
|
12 |
+
<input type="submit" value="PREDICT">
|
13 |
+
</form>
|
14 |
+
|
15 |
+
</body>
|
16 |
+
</html>
|
word2vec_65_epochs.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:bf74def542679e89e7db1bcd98e6a3a45102876fd46824ba216a99e800e1d042
|
3 |
+
size 193302192
|