Joshleeave's picture
Update app.py
ab20612 verified
raw
history blame contribute delete
No virus
1.86 kB
import gradio as gr
import requests
import io
from PIL import Image
import os
API_URL = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-base-1.0"
def query(payload):
auth_hf_api_token = os.environ.get("AUTH_HF_API_TOKEN")
authorization = "Bearer " + auth_hf_api_token
headers = {"Authorization": authorization}
response = requests.post(API_URL, headers=headers, json=payload)
return response.content
def genImage(character_name, description_of_the_character):
input = "Create a movie poster for " + character_name + "," + description_of_the_character + ",Disney Pixar movie style"
image_bytes = query({
"inputs": input,
})
image = Image.open(io.BytesIO(image_bytes))
return image
theme = gr.themes.Soft(radius_size=gr.themes.sizes.radius_none).set(body_background_fill="white")
demo = gr.Interface(genImage,
title="Bing Image Creator",
description="AI disney poster creator by https://www.bingimagecreator.cc",
inputs=[
gr.Textbox(lines=1, placeholder="Name your character",label="character name"),
gr.Textbox(lines=4, placeholder="Describe your character",label="description of your character")
], outputs=["image"], examples=[
["Mickey Mouse", "An anthropomorphic mouse who typically wears red shorts, large yellow shoes, and white gloves, Mickey is one of the world's most recognizable fictional characters."],
["Donald Duck", "An anthropomorphic white duck with a yellow-orange bill, legs, and feet. He typically wears a sailor shirt and cap with a bow tie."],
["Milo", "A curious and adventurous young scientist who dreams of exploring the furthest reaches of the galaxy."]
], theme=theme).queue(concurrency_count=100, api_open=False).launch(show_api=False, show_error=False)
demo.launch()