emoji-express 0.1.0__tar.gz

emoji_express-0.1.0/PKG-INFO

@@ -0,0 +1,25 @@
+Metadata-Version: 2.4
+Name: emoji-express
+Version: 0.1.0
+Summary: Lightweight Spanish text-to-emoji predictor using Sentence Transformers.
+Author: anndersoon
+Description-Content-Type: text/markdown
+Requires-Dist: joblib
+Requires-Dist: numpy
+Requires-Dist: huggingface_hub
+Requires-Dist: sentence-transformers
+Requires-Dist: scikit-learn
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: summary
+
+# emoji-express 🚀
+
+Lightweight Spanish text-to-emoji multi-label predictor using Sentence Transformers and Scikit-learn, hosted on Hugging Face Hub.
+
+## Installation
+
+```bash
+pip install emoji-express

emoji_express-0.1.0/README.md

@@ -0,0 +1,8 @@
+# emoji-express 🚀
+
+Lightweight Spanish text-to-emoji multi-label predictor using Sentence Transformers and Scikit-learn, hosted on Hugging Face Hub.
+
+## Installation
+
+```bash
+pip install emoji-express

emoji_express-0.1.0/emoji_express/__init__.py

@@ -0,0 +1,3 @@
+from ..predictor import EmojiPredictor
+
+__all__ = ["EmojiPredictor"]

emoji_express-0.1.0/emoji_express/predictor.py

@@ -0,0 +1,30 @@
+import joblib
+import numpy as np
+from huggingface_hub import hf_hub_download
+from sentence_transformers import SentenceTransformer
+
+class EmojiPredictor:
+    def __init__(self, repo_id: str = "famoso7/emoji-classifier-es", filename: str = "emoji_model_deep.bin"):
+        """Descarga e inicializa el modelo desde Hugging Face Hub."""
+        print(f"Loading model '{repo_id}'...")
+        model_path = hf_hub_download(repo_id=repo_id, filename=filename)
+        
+        artifacts = joblib.load(model_path)
+        self.neural_net = artifacts["neural_net"]
+        self.mlb = artifacts["mlb"]
+        self.embedder = SentenceTransformer(artifacts["embedder_name"])
+
+    def predict(self, text: str, top_k: int = 2, threshold: float = 0.3) -> str:
+        """Predice los emojis más probables para el texto ingresado."""
+        text_embedding = self.embedder.encode([text])
+        probs = self.neural_net.predict_proba(text_embedding)[0]
+
+        top_indices = np.argsort(probs)[::-1][:top_k]
+        predicted_emojis = [
+            self.mlb.classes_[i] for i in top_indices if probs[i] >= threshold
+        ]
+
+        if not predicted_emojis:
+            predicted_emojis = [self.mlb.classes_[top_indices[0]]]
+
+        return " ".join(predicted_emojis)

emoji_express-0.1.0/emoji_express.egg-info/PKG-INFO

@@ -0,0 +1,25 @@
+Metadata-Version: 2.4
+Name: emoji-express
+Version: 0.1.0
+Summary: Lightweight Spanish text-to-emoji predictor using Sentence Transformers.
+Author: anndersoon
+Description-Content-Type: text/markdown
+Requires-Dist: joblib
+Requires-Dist: numpy
+Requires-Dist: huggingface_hub
+Requires-Dist: sentence-transformers
+Requires-Dist: scikit-learn
+Dynamic: author
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: requires-dist
+Dynamic: summary
+
+# emoji-express 🚀
+
+Lightweight Spanish text-to-emoji multi-label predictor using Sentence Transformers and Scikit-learn, hosted on Hugging Face Hub.
+
+## Installation
+
+```bash
+pip install emoji-express

emoji_express-0.1.0/emoji_express.egg-info/SOURCES.txt

@@ -0,0 +1,9 @@
+README.md
+setup.py
+emoji_express/__init__.py
+emoji_express/predictor.py
+emoji_express.egg-info/PKG-INFO
+emoji_express.egg-info/SOURCES.txt
+emoji_express.egg-info/dependency_links.txt
+emoji_express.egg-info/requires.txt
+emoji_express.egg-info/top_level.txt

emoji_express-0.1.0/emoji_express.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

emoji_express-0.1.0/emoji_express.egg-info/requires.txt

@@ -0,0 +1,5 @@
+joblib
+numpy
+huggingface_hub
+sentence-transformers
+scikit-learn

emoji_express-0.1.0/emoji_express.egg-info/top_level.txt

@@ -0,0 +1 @@
+emoji_express

emoji_express-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

emoji_express-0.1.0/setup.py

@@ -0,0 +1,18 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="emoji-express",
+    version="0.1.0",
+    packages=find_packages(),
+    install_requires=[
+        "joblib",
+        "numpy",
+        "huggingface_hub",
+        "sentence-transformers",
+        "scikit-learn"
+    ],
+    author="anndersoon",
+    description="Lightweight Spanish text-to-emoji predictor using Sentence Transformers.",
+    long_description=open("README.md", encoding="utf-8").read(),
+    long_description_content_type="text/markdown",
+)