Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image
 import io
 import os
 
-# Set up your Hugging Face API key
 token = os.getenv("HF_TOKEN")
 
 API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
@@ -24,14 +23,12 @@ def generate_image(prompt):
     return None
 
 def main():
-    st.title("Stable Diffusion Image Generator")
+    st.title("Stable Diffusion XL 1.0")
 
-    # Get user input
     prompt = st.text_input("Enter a prompt for image generation:")
     
     if st.button("Generate Image"):
         if prompt:
-            # Generate and display the image
             image = generate_image(prompt)
             if image:
                 st.image(image, caption="Generated Image")