Spaces:

zoharzaig
/

emoji-predictor-space

Sleeping

zoharzaig commited on Sep 19

Commit

98a9f76

1 Parent(s): 6b10ce1

Dataset with more emojis

Files changed (2) hide show

emojis_dataset.json CHANGED Viewed

The diff for this file is too large to render. See raw diff

model.py CHANGED Viewed

@@ -13,7 +13,7 @@ class EmojiPredictor:
     def _vectorize_descriptions(self):
         # Get the sentence embedding for each description
-        descriptions = [item['original_description'] for item in self.emoji_data]
         return self.model.encode(descriptions)
     def predict(self, text):

     def _vectorize_descriptions(self):
         # Get the sentence embedding for each description
+        descriptions = [item['description'] for item in self.emoji_data]
         return self.model.encode(descriptions)
     def predict(self, text):