Update app.py

app.py

@@ -4,4 +4,6 @@ import streamlit as st
 
 pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
 pipeline.to("cuda")
-pipeline("An image of a squirrel in Picasso style").images[0]
+pipeline("An image of a squirrel in Picasso style").images[0]
+
+st.image(pipeline)