Alberto Carmona commited on
Commit
ea753ea
1 Parent(s): 6d1b898

Change the style of the generated image

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -163,7 +163,7 @@ def generate_text_from_image(img):
163
  def generate_drawing_from_image(img, steps=100, seed=42, guidance_scale=6.0):
164
  print('RUN: generate_drawing_from_image')
165
  caption = generate_text_from_image(img)
166
- caption = caption + ', style drawing of kid'
167
  print('\tcaption: ' + caption)
168
  gen_image = generate_image_from_text(caption, steps=steps, seed=seed, guidance_scale=guidance_scale)
169
  return gen_image
 
163
  def generate_drawing_from_image(img, steps=100, seed=42, guidance_scale=6.0):
164
  print('RUN: generate_drawing_from_image')
165
  caption = generate_text_from_image(img)
166
+ caption = "a kid's drawing of " + caption
167
  print('\tcaption: ' + caption)
168
  gen_image = generate_image_from_text(caption, steps=steps, seed=seed, guidance_scale=guidance_scale)
169
  return gen_image