defaults

app.py

@@ -66,6 +66,31 @@ def edict(x, source_text, edit_text,
     # x = decode_image(response)
     # return np.array(x)
 
+examples =
+    [
+        ['square_ims/american_gothic.jpg', 'A painting of two people frowning', 'A painting of two people smiling', 0.5, 3],
+        ['square_ims/colloseum.jpg', 'An old ruined building', 'A new modern office building', 0.8, 3],
+        ['square_ims/hf.png', 'Emoji', 'Emoji with a mustache', 0.8, 3],
+    ]
+
+for dog_i in [1, 2]:
+    for breed in ['Golden Retriever', 'Chihuahua', 'Dalmatian']:
+        examples.append([f'square_ims/imagenet_dog_{dog_i}.jpg', 'A dog', f'A {breed}', 0.8, 3])
+
+examples.append(['square_ims/scream.jpg', 'A painting of someone screaming', 'A painting of an alien', 0.5, 3])
+examples.append('square_ims/yosemite.jpg', 'Granite forest valley', 'Granite desert valley', 0.8, 3])
+examples.append('square_ims/einstein.jpg', 'Mouth open', 'Mouth closed', 0.8, 3])
+examples.append('square_ims/einstein.jpg', 'A man', 'A man in K.I.S.S. facepaint', 0.8, 3])
+examples.extend([
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A Chinese New Year cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A Union Jack cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A Nigerian flag cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A Santa Claus cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'An Easter cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A hedgehog cupcake', 0.8, 3],
+        ['square_ims/imagenet_cake_2.jpg', 'A cupcake', 'A rose cupcake', 0.8, 3],
+    ])
+
 iface = gr.Interface(fn=edict, inputs=["image",
                                        gr.Textbox(label="Original Description"),
                                        gr.Textbox(label="Edit Description"),
@@ -74,5 +99,6 @@ iface = gr.Interface(fn=edict, inputs=["image",
                                        gr.Slider(0.0, 1, value=0.8, step=0.05),
                                        gr.Slider(0, 10, value=3, step=0.5),
                                       ],
+                     examples = examples,
                      outputs="image")
 iface.launch()