Update app.py

app.py

@@ -3,7 +3,20 @@ import streamlit as st
 x = st.slider('Select a value')
 st.write(x, 'squared is', x * x)
 
-print("hello")
+from diffusers import DiffusionPipeline
+import torch
+
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
+
+
+# if using torch < 2.0
+# pipe.enable_xformers_memory_efficient_attention()
+
+prompt = "An astronaut riding a green horse"
+
+images = pipe(prompt=prompt).images[0]
+
+#print("hello")
 
 """
 import json