Update app.py

app.py

@@ -15,13 +15,10 @@ def predict(prompt):
     """ this function accepts a prompt and generate a 
     corresponding image for the given prompt  """
 
-    image = pipeline(prompt=prompt) #.images[0]
+    image = pipeline(prompt=prompt).images[0]
 
     return image
 
-image = predict(" a man that wears a cat")
-image.show()
-
 ### 4. Gradio app ###
 
 # Create title, description and article strings