Add title and modify the caption

app.py

@@ -174,6 +174,8 @@ def generate_text_from_image(img):
 def generate_drawing_from_image(img, steps=100, seed=42, guidance_scale=6.0):
     print('RUN: generate_drawing_from_image')
     caption = generate_text_from_image(img)
+    caption = caption + ', style drawing of kid'
+    print('\tcaption: ' + caption)
     gen_image = generate_image_from_text(caption, steps=steps, seed=seed, guidance_scale=guidance_scale)
     return gen_image
 
@@ -182,6 +184,7 @@ random_seed = random.randint(0, 2147483647)
 
 gr.Interface(
     generate_drawing_from_image,
+    title='Reimagine the same image but drawn by a kid :)',
     inputs=[
         gr.Image(type="pil"),
         gr.inputs.Slider(1, 100, label='Inference Steps', default=50, step=1),