Update app.py

app.py

@@ -1,10 +1,8 @@
-import streamlit as st
-from transformers import pipeline
+import torch
+from diffusers import StableDiffusionPipeline
 
-pipe = pipeline('sentiment-analysis')
+pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
+pipe = pipe.to("cuda")
 
-text =st.text_area('enter some text!')
-
-if text:
-    out = pipe(text)
-    st.json(out)
+prompt = "a photo of an astronaut riding a horse on mars"
+image = pipe(prompt).images[0]