Update app.py

app.py

@@ -5,31 +5,29 @@ 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!")
+#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):
+def generate_image(input_text):
     # Load Diffusion pipeline
-    pipe = StableDiffusionPipeline.from_pretrained("fluently/Fluently-XL-v2",torch_dtype=torch.float16)
-    
+    pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5",torch_dtype=torch.float16)
     # Generate image from text prompt
+    prompt = input_text
     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 App")
 # Text input for prompt
-prompt = st.text_input("Enter your image description:", "")
-
+input_text = st.text_input("Enter your image description:", "")
 # Button to generate image
 if st.button("Generate Image"):
     # Generate image based on the prompt
-    generated_image = generate_image(prompt)
+    st.info(input_text)
+    img = generate_image(input_text)
     
     # Display the generated image
-    st.image(generated_image, caption="Generated Image", use_column_width=True)
+    st.image(img, caption="Generated Image")