cpu and updated info

app.py

@@ -15,7 +15,7 @@ from PIL import Image, ImageOps
 
 # Constants
 TARGET_SIZE = (512, 512)
-DEVICE = torch.device("cuda")
+DEVICE = torch.device("cpu")
 LPIPS_MODELS = ['alex', 'vgg', 'squeeze']
 MASK_SIZES = {"64x64": 64, "128x128": 128, "256x256": 256}
 DEFAULT_MASK_SIZE = "256x256"
@@ -41,7 +41,7 @@ AUTHORS = """
     Pedro Reviriego<sup>1</sup>
 </div>
 <div align="center"; style="font-size: 1.3em; font-style: italic;">
-    <sup>1</sup>Universidad Politécnica de Madrid, <sup>2</sup>Universidad Carlos III de Madrid, <sup>3</sup>Universidad Antonio de Nebrija, <sup>4</sup>Universidad de Valladolid
+    <sup>1</sup>ETSI de Telecomunicación, Universidad Politécnica de Madrid, <sup>2</sup>Universidad Carlos III de Madrid, <sup>3</sup>Universidad Antonio de Nebrija, <sup>4</sup>Universidad de Valladolid
 </div>
 </body>
 """
@@ -80,6 +80,7 @@ BUTTONS = """
     <i class="fa-regular fa-folder-open"></i> Zenodo
 </a>
 </div>
+<center><img src="/file=static/greco_example.png" width="800" style="margin-top: 20px; border-radius: 15px;"></center>
 """
 
 DESCRIPTION = """
@@ -87,6 +88,8 @@ DESCRIPTION = """
 
 Welcome to our official demo for our [research paper](https://arxiv.org/abs/2407.09549) presented at the KDD conference workshop on [Evaluation and Trustworthiness of Generative AI Models](https://genai-evaluation-kdd2024.github.io/genai-evalution-kdd2024/).
 
+This demo shows the effects of recursively applying inpainting with a random mask to an image. A mask is applied at each iteration to remove a random part of the image and subsequently, inpainting is used to reconstruct the image. As iterations progress, the image can change significantly. You can see the effects of two iterations on "The Nobleman with his Hand on his Chest" by El Greco. Now is your turn, play with images, mask sizes, and iterations to see the effects of recursive inpainting!
+
 ## 🚀 How to Use
 
 1. 📤 Upload an image or choose from our examples from the [WikiArt dataset](https://huggingface.co/datasets/huggan/wikiart) used in our paper.
@@ -189,6 +192,7 @@ def inpaint_image(prompt: str, image: Image.Image, mask_image: Image.Image) -> I
 
 def create_gradio_interface():
     with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Default(primary_hue="red", secondary_hue="blue")) as demo:
+        gr.set_static_paths(paths=["static"])
         gr.Markdown(TITLE)
         gr.Markdown(AUTHORS)
         gr.HTML(BUTTONS)
@@ -246,4 +250,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.launch()
+    demo.launch(allowed_paths=["static"])