Update app.py

app.py

@@ -1,30 +1,30 @@
 import streamlit as st
 from PIL import Image
-from diffusers import DiffusionPipeline
-
-choice = st.sidebar.selectbox("Enter Your Choice",["Home","DALL-E","Diffusers"])
-if choice == "Home":
-    st.title("Text to Image Generation App!")
-elif choice == "DALL-E"
-    st.subheader("Text to image generation with DALL-E")
-else:
-    st.subheader("Text to image generation with Diffusers")
+import torch 
+from diffusers import StableDiffusionPipeline
 
+#choice = st.sidebar.selectbox("Enter Your Choice",["Home","DALL-E","Diffusers"])
+#if choice == "Home":
+  st.title("Text to Image Generation App!")
+#elif choice == "DALL-E"
+ #   st.subheader("Text to image generation with DALL-E")
+#else:
+ #   st.subheader("Text to image generation with Diffusers")
 
 def generate_image(prompt):
     # Load Diffusion pipeline
-    pipeline = DiffusionPipeline.from_pretrained("fluently/Fluently-XL-v2")
+    pipe = StableDiffusionPipeline.from_pretrained("fluently/Fluently-XL-v2",torch_dtype=torch.float16)
     
     # Generate image from text prompt
-    generated_image = pipeline(prompt)
+    generated_image = pipe(prompt).images[0]
     
     return generated_image
 
 # Set Streamlit app title
-st.title("Text to Image Generation")
+#st.title("Text to Image Generation")
 
 # Text input for prompt
-prompt = st.text_input("Enter your text prompt:", "")
+prompt = st.text_input("Enter your image description:", "")
 
 # Button to generate image
 if st.button("Generate Image"):