Update app.py

app.py

@@ -4,6 +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]
+image = pipeline("An image of a squirrel in Picasso style").images[0]
 
-st.image(pipeline)
+st.image(image, caption="Image")