try with image generation

app.py

@@ -1,10 +1,10 @@
 import streamlit as st
 
-# from diffusers import StableDiffusionPipeline
-# import torch
+from diffusers import StableDiffusionPipeline
+import torch
 
-# model_id = "runwayml/stable-diffusion-v1-5"
-# pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
+model_id = "runwayml/stable-diffusion-v1-5"
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 
 # pipe = pipe.to("cuda")
 
@@ -14,10 +14,11 @@ import streamlit as st
 
 
 with st.form("tti_form"):
-   st.write("Text to Image.")
-   prompt = st.text_input("Enter Prompt")
-   # Every form must have a submit button.
-   submitted = st.form_submit_button("Submit")
-   if submitted:
-       st.write("prompt", prompt)
-
+    st.write("Text to Image")
+    prompt = st.text_input("Enter Prompt")
+    # Every form must have a submit button.
+    submitted = st.form_submit_button("Submit")
+    if submitted:
+        st.write("prompt", prompt)
+        image = pipe(prompt).images[0]
+        st.image(image)

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+diffusers[torch]