Add sample images

app.py

@@ -13,23 +13,26 @@ DESCRIPTION = 'Expected execution time on Hugging Face Spaces: 5s (DDIM, 20 step
 FOOTER = '<img id="visitor-badge" src="https://visitor-badge.glitch.me/badge?page_id=hysts.diffusers-anime-faces" alt="visitor badge" />'
 
 
-def get_sample_image_url(name: str) -> str:
+def get_sample_image_url(file_name: str) -> str:
     sample_image_dir = 'https://huggingface.co/spaces/hysts/diffusers-anime-faces/resolve/main/samples'
-    return f'{sample_image_dir}/{name}.png'
+    return f'{sample_image_dir}/{file_name}'
 
 
 def get_sample_image_markdown(name: str) -> str:
     model_name = name.split()[0]
-    url = get_sample_image_url(model_name)
     if name == 'ddpm-128-exp000 (DDPM)':
-        text = f'''
-                - size: 128x128
-                - seed: 0-99
-                - scheduler: DDPM
-
-                ![sample images]({url})'''
+        file_name = f'{model_name}.png'
+    elif name == 'ddpm-128-exp000 (DDIM, 20 steps)':
+        file_name = f'{model_name}-ddim-20steps.png'
     else:
         raise ValueError
+    url = get_sample_image_url(file_name)
+    text = f'''
+            - size: 128x128
+            - seed: 0-99
+            - scheduler: DDPM
+
+            ![sample images]({url})'''
     return text
 
 
@@ -90,10 +93,10 @@ def main():
                 with gr.Row():
                     model_name2 = gr.Dropdown([
                         'ddpm-128-exp000 (DDPM)',
+                        'ddpm-128-exp000 (DDIM, 20 steps)',
                     ],
                                               value='ddpm-128-exp000 (DDPM)',
-                                              label='Model',
-                                              interactive=False)
+                                              label='Model')
                 with gr.Row():
                     text = get_sample_image_markdown(model_name2.value)
                     sample_images = gr.Markdown(text)