update documentation

app.py

@@ -1,47 +1,33 @@
-import gradio as gr
-import torch
 import os
 
-from PIL import Image
+import torch
+import boto3
+
+import gradio as gr
 import pandas as pd
 
 from transformers import CLIPProcessor, CLIPModel
 
 checkpoint = "vincentclaes/emoji-predictor"
-# x_, _, files = next(os.walk("./emojis"))
-# no_of_emojis = range(len(files))
-# emojis_as_images = [Image.open(f"emojis/{i}.png") for i in no_of_emojis]
 adjectives = pd.read_table("./adjectives.txt", header=None)[0].to_list()
 K = 4
+APP_NAME = "emoji-tagging"
+BUCKET = "drift-pilot-ml-model"
 
 processor = CLIPProcessor.from_pretrained(checkpoint)
 model = CLIPModel.from_pretrained(checkpoint)
 
 
-def concat_images(*images):
-    """Generate composite of all supplied images.
-    https://stackoverflow.com/a/71315656/1771155
-    """
-    # Get the widest width.
-    width = max(image.width for image in images)
-    # Add up all the heights.
-    height = max(image.height for image in images)
-    # set the correct size of width and heigtht of composite.
-    composite = Image.new("RGB", (2 * width, 2 * height))
-    assert K == 4, "We expect 4 suggestions, other numbers won't work."
-    for i, image in enumerate(images):
-        if i == 0:
-            composite.paste(image, (0, 0))
-        elif i == 1:
-            composite.paste(image, (width, 0))
-        elif i == 2:
-            composite.paste(image, (0, height))
-        elif i == 3:
-            composite.paste(image, (width, height))
-    return composite
-
-
-def get_tag(emoji, tags="", model=model, processor=processor, K=4):
+def log_inference():
+    if os.environ["CLIENT"]:
+        boto3.client("s3").put_object(
+            Body=more_binary_data,
+            Bucket=BUCKET,
+            Key=f"${APP_NAME}/",
+        )
+
+
+def get_tag(emoji, tags="", expected="", model=model, processor=processor, K=K):
     if tags:
         tags = tags.strip().split(",")
     else:
@@ -62,8 +48,12 @@ def get_tag(emoji, tags="", model=model, processor=processor, K=4):
 
 title = "Tagging an Emoji"
 description = """You provide an Emoji and our few-shot fine tuned CLIP model will suggest some tags that are appropriate.\n
-We use the 228 most common adjectives in english](https://grammar.yourdictionary.com/parts-of-speech/adjectives/list-of-adjective-words.html).\n
-You can also specify your own custom tags, for example; love,hate,fun,bitterness.
+
+- We use the [228 most common adjectives in english](https://grammar.yourdictionary.com/parts-of-speech/adjectives/list-of-adjective-words.html).\n
+- You can also specify your own custom tags. try with; 
+
+        love,hate,fun,bitterness,passion,dying,bliss
+
 """
 
 examples = [[f"emojis/{i}.png"] for i in range(32)]
@@ -71,18 +61,21 @@ examples = [[f"emojis/{i}.png"] for i in range(32)]
 text = gr.inputs.Textbox(
     placeholder="Enter a text and we will try to predict an emoji..."
 )
-gr.Interface(
+app = gr.Interface(
     fn=get_tag,
     inputs=[
         gr.components.Image(type="pil", label="emoji"),
         gr.components.Textbox(
             label="tags",
             placeholder="Provide a comma seperated list of tags; tag1,tag2,tag3,...",
-        ),
+        )
     ],
     outputs=gr.Textbox(),
     examples=examples,
     examples_per_page=32,
     title=title,
     description=description,
-).launch()
+)
+
+if __name__ == "__main__":
+    app.launch()

pyproject.toml

@@ -1,18 +1,23 @@
 [tool.poetry]
-name = "emoji-predictor"
+name = "emoji-tagging"
 version = "0.1.0"
 description = ""
-authors = ["vincent <vclaes1986@gmail.com>"]
+authors = ["Vincent Claes <vincent.v.claes@gmail.com>"]
+readme = "README.md"
+packages = [{include = "emoji_tagging"}]
 
 [tool.poetry.dependencies]
-python = "^3.9"
+python = "^3.10"
 torch = "^1.12.1"
-gradio = "^3.3.1"
-transformers = "^4.22.1"
-more-itertools = "^8.14.0"
+transformers = "^4.22.2"
+gradio = "^3.4.1"
+boto3 = "^1.24.89"
 
-[tool.poetry.dev-dependencies]
+
+
+[tool.poetry.group.dev.dependencies]
+black = "^22.10.0"
 
 [build-system]
-requires = ["poetry-core>=1.0.0"]
+requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -1,12 +1,13 @@
-aiohttp==3.8.1
+aiohttp==3.8.3
 aiosignal==1.2.0
-analytics-python==1.4.0
 anyio==3.6.1
 async-timeout==4.0.2
 attrs==22.1.0
-backoff==1.10.0
 bcrypt==4.0.0
-certifi==2022.9.14
+black==22.10.0
+boto3==1.24.89
+botocore==1.27.89
+certifi==2022.9.24
 cffi==1.15.1
 charset-normalizer==2.1.1
 click==8.1.3
@@ -16,33 +17,35 @@ cycler==0.11.0
 fastapi==0.85.0
 ffmpy==0.3.0
 filelock==3.8.0
-fonttools==4.37.2
+fonttools==4.37.4
 frozenlist==1.3.1
 fsspec==2022.8.2
-gradio==3.3.1
+gradio==3.4.1
 h11==0.12.0
 httpcore==0.15.0
 httpx==0.23.0
-huggingface-hub==0.9.1
+huggingface-hub==0.10.0
 idna==3.4
 Jinja2==3.1.2
+jmespath==1.0.1
 kiwisolver==1.4.4
 linkify-it-py==1.0.3
 markdown-it-py==2.1.0
 MarkupSafe==2.1.1
-matplotlib==3.6.0
-mdit-py-plugins==0.3.0
+matplotlib==3.6.1
+mdit-py-plugins==0.3.1
 mdurl==0.1.2
-monotonic==1.6
-more-itertools==8.14.0
 multidict==6.0.2
+mypy-extensions==0.4.3
 numpy==1.23.3
 orjson==3.8.0
 packaging==21.3
-pandas==1.4.4
+pandas==1.5.0
 paramiko==2.11.0
+pathspec==0.10.1
 Pillow==9.2.0
 pip==22.2.2
+platformdirs==2.5.2
 pycparser==2.21
 pycryptodome==3.15.0
 pydantic==1.10.2
@@ -51,12 +54,13 @@ PyNaCl==1.5.0
 pyparsing==3.0.9
 python-dateutil==2.8.2
 python-multipart==0.0.5
-pytz==2022.2.1
+pytz==2022.4
 PyYAML==6.0
 regex==2022.9.13
 requests==2.28.1
 rfc3986==1.5.0
-setuptools==62.6.0
+s3transfer==0.6.0
+setuptools==65.4.1
 setuptools-scm==7.0.5
 six==1.16.0
 sniffio==1.3.0
@@ -65,8 +69,8 @@ tokenizers==0.12.1
 tomli==2.0.1
 torch==1.12.1
 tqdm==4.64.1
-transformers==4.22.1
-typing_extensions==4.3.0
+transformers==4.22.2
+typing_extensions==4.4.0
 uc-micro-py==1.0.1
 urllib3==1.26.12
 uvicorn==0.18.3