add mapping for artist

app.py

@@ -1,8 +1,12 @@
 import os
+import json
 import requests
 
 import gradio as gr
 
+with open("mapping.json", "r") as f:
+    mapping = json.load(f)
+
 def get_images(text):
     headers = {
         "Content-Type": "application/json",
@@ -19,12 +23,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 = [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 = """
         .gradio-container {
@@ -198,7 +205,7 @@ with block:
 
         gallery = gr.Gallery(
             label="Art Pieces", show_label=False, elem_id="gallery"
-        ).style(grid=[2], height="auto")
+        ).style(grid=[2], height="auto", container=True)
         btn.click(get_images, inputs=text, outputs=gallery, postprocess=False)
        
         gr.HTML(