Update app.py

app.py

@@ -2,18 +2,24 @@ import streamlit as st
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Load the DALL-E model and tokenizer
-model_id = "runwayml/stable-diffusion-v1-5"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+# Load the model
+repo = "IDKiro/sdxs-512-0.9"
+seed = 42
+pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float32)
 
 # Streamlit app
 st.title("Image generation demo")
 
-prompt = st.text_input("Enter a text prompt:", "a cat in the rain")
+prompt = st.text_input("Enter a text prompt:", "A gold ring")
 
 if st.button("Generate Image"):
     # Generate image from the prompt
-    image = pipe(prompt).images[0]  
+    image = pipe(
+        prompt, 
+        num_inference_steps=1, 
+        guidance_scale=0,
+        generator=torch.manual_seed(seed)
+    ).images[0]
     
     # Display the image
     st.image(image, caption="Generated Image", channels="RGB")