Clean up

app.py

@@ -13,48 +13,6 @@ 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(file_name: str) -> str:
-    sample_image_dir = 'https://huggingface.co/spaces/hysts/diffusers-anime-faces/resolve/main/samples'
-    return f'{sample_image_dir}/{file_name}'
-
-
-def get_sample_image_markdown(name: str) -> str:
-    model_name = name.split()[0]
-    if name == 'ddpm-128-exp000 (DDPM)':
-        scheduler = 'DDPM'
-        steps = 1000
-        file_name = f'{model_name}.png'
-    elif name == 'ddpm-128-exp000 (DDIM, 20 steps)':
-        scheduler = 'DDIM'
-        steps = 20
-        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: {scheduler}
-            - steps: {steps}
-
-            ![sample images]({url})'''
-    return text
-
-
-def update_scheduler_type(name: str) -> dict:
-    visible = name != 'DDPM'
-    if name == 'PNDM':
-        minimum = 4
-        maximum = 100
-    else:
-        minimum = 1
-        maximum = 200
-    return gr.Slider.update(visible=visible,
-                            minimum=minimum,
-                            maximum=maximum,
-                            value=20)
-
-
 def create_simple_demo(model: Model) -> gr.Blocks:
     with gr.Blocks() as demo:
         run_button = gr.Button('Generate')
@@ -64,6 +22,19 @@ def create_simple_demo(model: Model) -> gr.Blocks:
 
 
 def create_advanced_demo(model: Model) -> gr.Blocks:
+    def update_scheduler_type(name: str) -> dict:
+        visible = name != 'DDPM'
+        if name == 'PNDM':
+            minimum = 4
+            maximum = 100
+        else:
+            minimum = 1
+            maximum = 200
+        return gr.Slider.update(visible=visible,
+                                minimum=minimum,
+                                maximum=maximum,
+                                value=20)
+
     with gr.Blocks() as demo:
         gr.Markdown(DESCRIPTION)
 
@@ -119,6 +90,32 @@ def create_advanced_demo(model: Model) -> gr.Blocks:
 
 
 def create_sample_image_view_demo() -> gr.Blocks:
+    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}/{file_name}'
+
+    def get_sample_image_markdown(name: str) -> str:
+        model_name = name.split()[0]
+        if name == 'ddpm-128-exp000 (DDPM)':
+            scheduler = 'DDPM'
+            steps = 1000
+            file_name = f'{model_name}.png'
+        elif name == 'ddpm-128-exp000 (DDIM, 20 steps)':
+            scheduler = 'DDIM'
+            steps = 20
+            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: {scheduler}
+                - steps: {steps}
+
+                ![sample images]({url})'''
+        return text
+
     with gr.Blocks() as demo:
         with gr.Row():
             model_name = gr.Dropdown([