Update app.py

app.py

@@ -3,31 +3,23 @@ from PIL import Image
 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(input_text):
     # Load Diffusion pipeline
-    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",torch_dtype=torch.float16)
+    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
     # Generate image from text prompt
-    pipe = pipe.to("cuda")
     prompt = input_text
     generated_image = pipe(prompt).images[0]
     return generated_image
     
 # Set Streamlit app title
 st.title("Text to Image Generation App")
+
 # Text input for prompt
 input_text = st.text_input("Enter your image description:", "")
+
 # Button to generate image
 if st.button("Generate Image"):
     # Generate image based on the prompt
-    st.info(input_text)
     img = generate_image(input_text)
     
     # Display the generated image