Update app.py

app.py

@@ -1,30 +1,26 @@
 import streamlit as st
+import requests
 from PIL import Image
-from transformers import AutoProcessor, KandinskyDecoderPipeline
-import torch
+from io import BytesIO
 
-@st.cache_resource
-def load_model():
-    processor = AutoProcessor.from_pretrained("kandinsky-community/kandinsky-2-2-decoder")
-    model = KandinskyDecoderPipeline.from_pretrained(
-        "kandinsky-community/kandinsky-2-2-decoder",
-        torch_dtype=torch.float32,
-    )
-    model.to("cpu")
-    return processor, model
+st.set_page_config(page_title="DALLE Mini Thumbnail Generator")
+st.title("🖼️ Thumbnail Generator using DALL·E Mini")
 
-st.set_page_config(page_title="Thumbnail AI - Kandinsky", layout="centered")
-st.title("🎨 Kandinsky Thumbnail Generator")
+prompt = st.text_input("Enter your thumbnail prompt:", "Minecraft NOOB vs PRO Hardcore battle")
 
-prompt = st.text_input("📝 Enter your thumbnail idea:", "Minecraft Hardcore NOOB vs PRO battle")
+if st.button("Generate"):
+    with st.spinner("Generating Image..."):
+        api_url = "https://api-inference.huggingface.co/models/dalle-mini/dalle-mini/mega-1-fp16:latest"
+        headers = {"Authorization": "Bearer YOUR_HF_API_TOKEN"}  # replace with real token
 
-if st.button("🎬 Generate Image"):
-    with st.spinner("Loading model and generating..."):
-        processor, model = load_model()
-        image = model(prompt=prompt).images[0]
-        st.image(image, caption="Generated Thumbnail", use_column_width=True)
-        st.success("✅ Done!")
+        response = requests.post(api_url, headers=headers, json={"inputs": prompt})
+        if response.status_code == 200:
+            output = response.json()
+            image_url = output["generated_images"][0]
+            image_data = requests.get(image_url).content
+            image = Image.open(BytesIO(image_data))
 
-        # Download button
-        img_bytes = image.convert("RGB").tobytes("raw", "RGB")
-        st.download_button("Download Image", data=img_bytes, file_name="thumbnail.jpg", mime="image/jpeg")
+            st.image(image, caption="Generated Thumbnail")
+            st.success("Done!")
+        else:
+            st.error("❌ Failed to generate image. Try again later.")