| import streamlit as st |
| import requests |
| from PIL import Image |
| from io import BytesIO |
|
|
| st.set_page_config(page_title="DALLE Mini Thumbnail Generator") |
| st.title("🖼️ Thumbnail Generator using DALL·E Mini") |
|
|
| prompt = st.text_input("Enter your thumbnail prompt:", "Minecraft NOOB vs PRO Hardcore 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"} |
|
|
| 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)) |
|
|
| st.image(image, caption="Generated Thumbnail") |
| st.success("Done!") |
| else: |
| st.error("❌ Failed to generate image. Try again later.") |
