add artist label

app.py

@@ -1,8 +1,12 @@
 import os
 import requests
+import json
 
 import gradio as gr
 
+with open("mapping.json", "r") as f:
+    mapping = json.load(f)
+
 
 def get_images(text):
     headers = {
@@ -20,12 +24,15 @@ def get_images(text):
         json={"data": text},
     )
     images = []
-    image_data = response.json()["image"]
+    response_json = response.json()
+    image_data = response_json["image"]
+    image_label = [f"Artist: {mapping[str(id_)]}" for id_ in response_json["id"]]
+
     for image in image_data:
         # got this from https://huggingface.co/spaces/stabilityai/stable-diffusion/blob/main/app.py
         image_b64 = f"data:image/jpeg;base64,{image}"
         images.append(image_b64)
-    return images
+    return tuple(zip(images, image_label))
 
 
 css = """
@@ -182,7 +189,7 @@ with block:
               </div>
               <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
                 Search the <a style="text-decoration: underline;" href="https://archive.org/details/wikiart-dataset">WikiArt dataset</a> containing 81 444 paintings. 
-                </br>Read the article <a style="text-decoration: underline;" href="https://medium.com/@vincentclaes_43752/a110a2f5bf8e?source=friends_link&sk=cabf93a94d582ebbc53d3672cbb76a98">Searching Art through Deep Learning</a>
+                </br>Read the article <a style="text-decoration: underline;" href="https://medium.com/@vincentclaes_43752/a110a2f5bf8e?source=friends_link&sk=cabf93a94d582ebbc53d3672cbb76a98">Searching Art with Artifical Intelligence</a>
               </p>
             </div>
         """
@@ -225,8 +232,22 @@ with block:
         gr.HTML(
             """
                 <div class="footer">
-                    <p>Model by <a href="https://www.meet-drift.ai/" style="text-decoration: underline;" target="_blank">Drift</a> 
+                    <p>Model by <a href="https://www.meet-drift.ai/" style="text-decoration: underline;" target="_blank">Drift</a> </p>
+                </div>
+                </br>
+                <div style="text-align: center; margin: 0 auto;">
+                    <p style="margin-bottom: 10px; font-size: 110%; line-height: 23px;">
+                    Want a free demo on a sample of your image data?
+                    </p>
+                    <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
+                    <a style="text-decoration: underline;" href="https://calendly.com/vincent-v-claes">Schedule a call</a> or contact me on  <a style="text-decoration: underline;" href="https://www.linkedin.com/in/vincent-claes-0b346337">Linkedin</a>.
+                    </p>
+                    <p style="margin-bottom: 10px; font-size: 66%; line-height: 23px; text-align: left; margin-left: 225px">
+                    > The images can be about anything! Art, animals, clothes,… </br>
+                    > Low-resolution pictures are fine.</br>
+                    > Watermarks on the pictures should be ok to validate the performance.</br>
                     </p>
+                    
                 </div>
            """
         )

requirements.txt

@@ -1,27 +1,28 @@
 aiohttp==3.8.3
 aiosignal==1.3.1
+altair==4.2.0
 anyio==3.6.2
 async-timeout==4.0.2
 attrs==22.1.0
-bcrypt==4.0.1
 certifi==2022.12.7
-cffi==1.15.1
 charset-normalizer==2.1.1
 click==8.1.3
 contourpy==1.0.6
-cryptography==38.0.4
 cycler==0.11.0
+entrypoints==0.4
 fastapi==0.88.0
 ffmpy==0.3.0
 fonttools==4.38.0
 frozenlist==1.3.3
 fsspec==2022.11.0
-gradio==3.12.0
-h11==0.12.0
-httpcore==0.15.0
+gradio==3.14.0
+h11==0.14.0
+httpcore==0.16.2
 httpx==0.23.1
 idna==3.4
+importlib-resources==5.10.1
 Jinja2==3.1.2
+jsonschema==4.17.3
 kiwisolver==1.4.4
 linkify-it-py==1.0.3
 markdown-it-py==2.1.0
@@ -34,32 +35,33 @@ numpy==1.23.5
 orjson==3.8.3
 packaging==22.0
 pandas==1.5.2
-paramiko==2.12.0
 Pillow==9.3.0
-pip==22.3.1
-pycparser==2.21
+pip==22.3
+pkgutil_resolve_name==1.3.10
 pycryptodome==3.16.0
 pydantic==1.10.2
 pydub==0.25.1
-PyNaCl==1.5.0
 pyparsing==3.0.9
+pyrsistent==0.19.2
 python-dateutil==2.8.2
 python-dotenv==0.21.0
 python-multipart==0.0.5
-pytz==2022.6
+pytz==2022.7
 PyYAML==6.0
 requests==2.28.1
 rfc3986==1.5.0
 setuptools==65.6.3
-setuptools-scm==7.0.5
+setuptools-scm==7.1.0
 six==1.16.0
 sniffio==1.3.0
 starlette==0.22.0
 tomli==2.0.1
+toolz==0.12.0
 typing_extensions==4.4.0
 uc-micro-py==1.0.1
 urllib3==1.26.13
 uvicorn==0.20.0
 websockets==10.4
-wheel==0.38.4
+wheel==0.37.1
 yarl==1.8.2
+zipp==3.11.0