Create app.py

app.py

@@ -0,0 +1,33 @@
+import streamlit as st
+from diffusers import StableDiffusionPipeline
+import torch
+from PIL import Image
+
+# Load model (run once, slow on CPU)
+@st.cache_resource
+def load_model():
+    model_id = "runwayml/stable-diffusion-v1-5"  # You can change to your own fine-tuned model
+    pipe = StableDiffusionPipeline.from_pretrained(
+        model_id,
+        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+    )
+    pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+    return pipe
+
+st.title("🖼️ AI Thumbnail Generator (Stable Diffusion)")
+
+# Load Stable Diffusion pipeline
+pipe = load_model()
+
+# UI
+prompt = st.text_input("📌 Enter Thumbnail Prompt", value="Minecraft NOOB vs PRO with fire background, bold text")
+
+if st.button("Generate Thumbnail"):
+    with st.spinner("Generating image..."):
+        image = pipe(prompt, guidance_scale=8.5).images[0]
+        st.image(image, caption="Generated Thumbnail", use_column_width=True)
+
+        # Option to download
+        img_bytes = image.save("thumbnail.jpg")
+        with open("thumbnail.jpg", "rb") as f:
+            st.download_button("📥 Download Thumbnail", f, "thumbnail.jpg", "image/jpeg")