kasper-boy commited on
Commit
246c2a2
·
verified ·
1 Parent(s): 554b0af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -35,6 +35,7 @@ def image_generation(prompt):
35
  def caption_my_image(pil_image):
36
  semantics = caption_image(images=pil_image)[0]['generated_text']
37
  images = image_generation(semantics)
 
38
 
39
  demo = gr.Interface(fn=caption_my_image,
40
  inputs=[gr.Image(label="Select Image",type="pil")],
 
35
  def caption_my_image(pil_image):
36
  semantics = caption_image(images=pil_image)[0]['generated_text']
37
  images = image_generation(semantics)
38
+ return images
39
 
40
  demo = gr.Interface(fn=caption_my_image,
41
  inputs=[gr.Image(label="Select Image",type="pil")],