example update & better descriptions

app.py

@@ -1,4 +1,5 @@
 from pathlib import Path
+from itertools import chain
 
 import gdown
 # from PIL import Image
@@ -14,7 +15,10 @@ assert Path('./data/esper_demo').is_dir()
 
 # example image from COCO data
 image_urls = {
-    '108953': 'https://farm8.staticflickr.com/7160/6484651991_9d1eaa557a_z.jpg'
+    # '108953': 'https://farm8.staticflickr.com/7160/6484651991_9d1eaa557a_z.jpg'
+    # '394905': 'http://farm5.staticflickr.com/4151/4955575345_6c6bdfae9d_z.jpg',
+    '330341': 'https://farm4.staticflickr.com/3018/3451080626_4a43435f4b_z.jpg',
+    '396820': 'https://farm1.staticflickr.com/148/351466274_8d7174e11b_z.jpg'
 }
 images = {}
 for k, url in image_urls.items():
@@ -32,12 +36,14 @@ for k, v in images.items():
         images[k] = image
 '''
 
+prompts = ['blog:', 'dialogue:', 'This is my favorite poem:']
 
-examples = [[
-    v,
-    'My favourite recipe:',
-    20,
-    False
-] for v in images.values()]
+title = 'Demo for ESPER'
+description = 'backbone: style-finetuned GPT-2-base'
+prompt_label = f'Prompt (try pretrained styles such as "blog:" or "dialogue:" or unseen prompts such as "{prompts[-1]}")'
 
-launch(examples)
+examples = [[[v, prompt, 20, False ] for prompt in prompts]
+            for v in images.values()]
+examples = list(chain(*examples))
+
+launch(examples, title=title, description=description, prompt_label=prompt_label)

arguments.py

@@ -20,7 +20,7 @@ def get_args():
     parser.add_argument(
         '--label_path', type=str, default='./data/esper_demo/labels_all.json', help='style label info file path')
     parser.add_argument(
-        '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style', help='checkpoint file path')
+        '--checkpoint', type=str, default='./data/esper_demo/ckpt/gpt2_style_2', help='checkpoint file path')
 
     parser.add_argument(
         '--prefix_length', type=int, default=10, help='prefix length for the visual mapper')

run.py

@@ -140,21 +140,22 @@ img, _, text = run(sample_img, 'There lies', 50, 20, sample=False)
 print('test_run:', text)
 '''
 
-def launch(examples=None):
+def launch(examples=None, title='Demo for ESPER', description=None, prompt_label='Prompt'):
     args = get_args()
     inferer = prepare(args)
     runner = Runner(inferer)
 
     iface = gr.Interface(
-        title="Demo for ESPER",
+        title=title,
+        description=description,
         fn=runner.__call__,
-        inputs=[gr.components.Image(shape=(224, 224)),
-                gr.components.Textbox(label='prompt'),
-                gr.components.Slider(20, 120, step=1, label='length'),
+        inputs=[gr.components.Image(shape=(224, 224), label='Image'),
+                gr.components.Textbox(label=prompt_label),
+                gr.components.Slider(20, 40, step=1, label='Length'),
                 # gr.components.Slider(10, 100, step=1, label='window_size'),
                 gr.components.Checkbox(label='do sample')],
-        outputs=[gr.components.Textbox(label='prompt'),
-                gr.components.Textbox(label='generation')],
+        outputs=[gr.components.Textbox(label='Prompt'),
+                gr.components.Textbox(label='Generation')],
         examples=examples
     )
     if args.port is not None: