Merge pull request #50 from barun-saha/visual

file_embeddings/embeddings.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4c2758d74daff23f638acff398d1a512c74b97178118b7adbe8433ae8d368f52
-size 158848
+oid sha256:64a1ba79b20c81ba7ed6604468736f74ae89813fe378191af1d8574c008b3ab5
+size 326784

file_embeddings/icons.npy

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2deede2276676fff6e70e585f5c05d004b3f24d5dc389a3c09228fbd814bcfce
-size 49728
+oid sha256:ce5ce4c86bb213915606921084b3516464154edcae12f4bc708d62c6bd7acebb
+size 51168

global_config.py

@@ -35,7 +35,7 @@ class GlobalConfig:
 
     LLM_PROGRESS_MAX = 90
     ICONS_DIR = 'icons/png128/'
-    TINY_BERT_MODEL = 'gaunernst/bert-tiny-uncased'
+    TINY_BERT_MODEL = 'gaunernst/bert-mini-uncased'
     EMBEDDINGS_FILE_NAME = 'file_embeddings/embeddings.npy'
     ICONS_FILE_NAME = 'file_embeddings/icons.npy'
 

helpers/icons_embeddings.py

@@ -115,9 +115,51 @@ def main():
     # Run this again if icons are to be added/removed
     save_icons_embeddings()
 
-    keywords = ['deep learning', 'library', 'universe', 'brain', 'cybersecurity', 'gaming', '']
+    keywords = [
+        'deep learning',
+        '',
+        'recycling',
+        'handshake',
+        'Ferry',
+        'rain drop',
+        'speech bubble',
+        'mental resilience',
+        'turmeric',
+        'Art',
+        'price tag',
+        'Oxygen',
+        'oxygen',
+        'Social Connection',
+        'Accomplishment',
+        'Python',
+        'XML',
+        'Handshake',
+    ]
     icon_files = find_icons(keywords)
-    print(f'The relevant icon files are: {icon_files}')
+    print(
+        f'The relevant icon files are:\n'
+        f'{list(zip(keywords, icon_files))}'
+    )
+
+    # BERT tiny:
+    # [('deep learning', 'deep-learning'), ('', '123'), ('recycling', 'refinery'),
+    #  ('handshake', 'dash-circle'), ('Ferry', 'cart'), ('rain drop', 'bucket'),
+    #  ('speech bubble', 'globe'), ('mental resilience', 'exclamation-triangle'),
+    #  ('turmeric', 'kebab'), ('Art', 'display'), ('price tag', 'bug-fill'),
+    #  ('Oxygen', 'radioactive')]
+
+    # BERT mini
+    # [('deep learning', 'deep-learning'), ('', 'compass'), ('recycling', 'tools'),
+    #  ('handshake', 'bandaid'), ('Ferry', 'cart'), ('rain drop', 'trash'),
+    #  ('speech bubble', 'image'), ('mental resilience', 'recycle'), ('turmeric', 'linkedin'),
+    #  ('Art', 'book'), ('price tag', 'card-image'), ('Oxygen', 'radioactive')]
+
+    # BERT small
+    # [('deep learning', 'deep-learning'), ('', 'gem'), ('recycling', 'tools'),
+    #  ('handshake', 'handbag'), ('Ferry', 'truck'), ('rain drop', 'bucket'),
+    #  ('speech bubble', 'strategy'), ('mental resilience', 'deep-learning'),
+    #  ('turmeric', 'flower'),
+    #  ('Art', 'book'), ('price tag', 'hotdog'), ('Oxygen', 'radioactive')]
 
 
 if __name__ == '__main__':

icons/svg_repo.txt

@@ -17,6 +17,7 @@ Icons collections used (and their licenses) from SVG Repo:
 - Objects Infographic Icons (CC0 License): https://www.svgrepo.com/collection/objects-infographic-icons/
 - Scientifics Study Collection (CC0 License): https://www.svgrepo.com/collection/scientifics-study/
 - Thanksgiving 4 Collection (CC0 License): https://www.svgrepo.com/collection/thanksgiving-4/
+- Using Hands Collection (CC0 License): https://www.svgrepo.com/collection/using-hands/
 - Vaadin Flat Vectors Collection (Apache License): https://www.svgrepo.com/collection/vaadin-flat-vectors/
 
 - India ICON SET (SVG) [NO attribution required]: https://icon666.com/collection/india_m7tgpnohh
@@ -172,3 +173,4 @@ https://www.svgrepo.com/download/423082/fast-food-fried-2.svg
 https://www.svgrepo.com/download/209887/tea-coffee-cup.svg
 https://www.svgrepo.com/download/209855/restaurant-spoon.svg
 https://www.svgrepo.com/download/209875/jelly-jar.svg
+https://www.svgrepo.com/download/83723/handshake.svg