|
|
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.") |
