updates

app.py

@@ -35,6 +35,7 @@ def poem_generate(word):
   return poem
 
 def poem_to_image(poem):
+  print(poem)
   poem = " ".join(poem[1:].split('\n'))
   poem = poem + " oil on canvas."
   steps, width, height, images, diversity = '50','256','256','1',15
@@ -46,7 +47,7 @@ demo = gr.Blocks()
 with demo:
   input_word = gr.Textbox(placeholder="Enter a word here to create a Poem on..")
   poem_txt = gr.Textbox(lines=7)
-  output_image = gr.Image(type="filepath")
+  output_image = gr.Image(type="filepath", shape=(256,256))
 
   b1 = gr.Button("Generate Poem")
   b2 = gr.Button("Generate Image")