Update index.html

index.html

@@ -3,7 +3,7 @@ from diffusers import StableDiffusionPipeline
 import torch
 
 # Load the model
-model_id = "runwayml/stable-diffusion-v2-1"
+model_id = "runwayml/stable-diffusion-v1-5"
 pipe = StableDiffusionPipeline.from_pretrained(model_id)
 pipe = pipe.to("cuda")  # Runs on Hugging Face's GPU servers
 
@@ -14,13 +14,4 @@ prompt = st.text_input("Describe your image...")
 if prompt:
     with st.spinner("Generating..."):
         image = pipe(prompt).images[0]
-        st.image(image, caption=prompt, use_column_width=True)
-        <button onclick="randomizePrompt()">Surprise Me</button>
-
-<script>
-  function randomizePrompt() {
-    const adjectives = ["cartoon", "realistic", "cyberpunk", "watercolor"];
-    const randomAdjective = adjectives[Math.floor(Math.random() * adjectives.length)];
-    document.getElementById("prompt").value = `${randomAdjective} dog`;
-  }
-</script>
+        st.image(image, caption=prompt, use_column_width=True)