|
import gradio as gr |
|
import requests |
|
from PIL import Image |
|
from io import BytesIO |
|
import base64 |
|
|
|
api_url = "https://5cb20b40-572c-426f-9466-995256f9b6eb.id.repl.co/generate_image" |
|
|
|
def generate_image(prompt, seed=0, negative_prompt="", model="Dreamlike Diffusion", sampler="k_dpmpp_2s_a", steps=50): |
|
data = "?prompt="+ prompt + "&seed="+ str(seed) + "&negative_prompt=" + negative_prompt + "&model=" + model + "&sampler=" + sampler + "&steps=" + str(steps) |
|
response = requests.post(api_url+data, timeout=400) |
|
if response.status_code == 200: |
|
img_base64 = response.json()["url"] |
|
img_bytes = base64.b64decode(img_base64) |
|
img = Image.open(BytesIO(img_bytes)) |
|
return img |
|
else: |
|
return None |
|
|
|
inputs = [ |
|
gr.inputs.Textbox(label="Prompt"), |
|
gr.inputs.Number(label="Seed", default=0), |
|
gr.inputs.Textbox(label="Negative Prompt", default=""), |
|
gr.inputs.Dropdown(['3DKX', 'Abyss OrangeMix', 'AbyssOrangeMix-AfterDark', 'ACertainThing', |
|
'AIO Pixel Art', 'Analog Diffusion', 'Anime Pencil Diffusion', 'Anygen', |
|
'Anything Diffusion', 'Anything v3', 'anything_v4_inpainting', |
|
'App Icon Diffusion', 'Arcane Diffusion', 'Archer Diffusion', |
|
'Asim Simpsons', 'A to Zovya RPG', 'Balloon Art', 'Borderlands', 'BPModel', |
|
'BubblyDubbly', 'Char', 'CharHelper', 'Cheese Daddys Landscape Mix', |
|
'ChilloutMix', 'ChromaV5', 'Classic Animation Diffusion', 'Clazy', |
|
'Colorful', 'Coloring Book', 'Comic-Diffusion', 'Concept Sheet', |
|
'Counterfeit', 'Cyberpunk Anime Diffusion', 'CyriousMix', |
|
'Dan Mumford Style', 'Darkest Diffusion', 'Dark Victorian Diffusion', |
|
'Deliberate', 'DGSpitzer Art Diffusion', 'Disco Elysium', 'DnD Item', |
|
'Double Exposure Diffusion', 'Dreamlike Diffusion', |
|
'dreamlike_diffusion_inpainting', 'Dreamlike Photoreal', |
|
'DreamLikeSamKuvshinov', 'Dreamshaper', 'DucHaiten', |
|
'DucHaiten Classic Anime', 'Dungeons and Diffusion', 'Dungeons n Waifus', |
|
'Eimis Anime Diffusion', 'Elden Ring Diffusion', "Elldreth's Lucid Mix", |
|
'Elldreths Retro Mix', 'Epic Diffusion', 'Eternos', 'Experience', |
|
'ExpMix Line', 'FaeTastic', 'Fantasy Card Diffusion', 'FKing SciFi', |
|
'Funko Diffusion', 'Furry Epoch', 'Future Diffusion', 'Ghibli Diffusion', |
|
'GorynichMix', 'Grapefruit Hentai', 'Graphic-Art', |
|
'GTA5 Artwork Diffusion', 'GuoFeng', 'Guohua Diffusion', 'HASDX', |
|
'Hassanblend', "Healy's Anime Blend", 'Hentai Diffusion', 'HRL', 'iCoMix', |
|
'Illuminati Diffusion', 'Inkpunk Diffusion', 'Jim Eidomode', |
|
'JWST Deep Space Diffusion', 'Kenshi', 'Knollingcase', 'Korestyle', |
|
'kurzgesagt', 'Laolei New Berry Protogen Mix', "Lawlas's yiff mix", |
|
'Liberty', 'Marvel Diffusion', 'Mega Merge Diffusion', 'Microcasing', |
|
'Microchars', 'Microcritters', 'Microscopic', 'Microworlds', |
|
'Midjourney Diffusion', 'Midjourney PaintArt', 'Min Illust Background', |
|
'ModernArt Diffusion', 'mo-di-diffusion', 'Moedel', 'MoistMix', |
|
'Movie Diffusion', 'NeverEnding Dream', 'Nitro Diffusion', 'Openniji', |
|
'OrbAI', 'Papercutcraft', 'Papercut Diffusion', 'Pastel Mix', |
|
'Perfect World', 'PFG', 'PIXHELL', 'Poison', 'Pokemon3D', 'PortraitPlus', |
|
'PPP', 'Pretty 2.5D', 'PRMJ', 'Project Unreal Engine 5', 'ProtoGen', |
|
'Protogen Anime', 'Protogen Infinity', 'Pulp Vector Art', 'PVC', |
|
'Rachel Walker Watercolors', 'Rainbowpatch', 'Ranma Diffusion', |
|
'RCNZ Dumb Monkey', 'RCNZ Gorilla With A Brick', 'RealBiter', |
|
'Realism Engine', 'Realistic Vision', 'Redshift Diffusion', 'Rev Animated', |
|
'Robo-Diffusion', 'Rodent Diffusion', 'RPG', 'Samdoesarts Ultmerge', |
|
'Sci-Fi Diffusion', 'SD-Silicon', 'Seek.art MEGA', 'Smoke Diffusion', |
|
'Something', 'Sonic Diffusion', 'Spider-Verse Diffusion', |
|
'Squishmallow Diffusion', 'stable_diffusion', 'stable_diffusion_2.1', |
|
'stable_diffusion_2_inpainting', 'Supermarionation', 'Sygil-Dev Diffusion', |
|
'Synthwave', 'SynthwavePunk', 'TrexMix', 'trinart', 'Trinart Characters', |
|
'Tron Legacy Diffusion', 'T-Shirt Diffusion', 'T-Shirt Print Designs', |
|
'Uhmami', 'Ultraskin', 'UMI Olympus', 'Unstable Ink Dream', 'URPM', |
|
'Valorant Diffusion', 'Van Gogh Diffusion', 'Vector Art', 'vectorartz', |
|
'Vintedois Diffusion', 'VinteProtogenMix', 'Vivid Watercolors', |
|
'Voxel Art Diffusion', 'waifu_diffusion', 'Wavyfusion', 'Woop-Woop Photo', |
|
'Xynthii-Diffusion', 'Yiffy', 'Zack3D', 'Zeipher Female Model', |
|
'Zelda BOTW'], label="Model", default="Dreamlike Diffusion"), |
|
gr.inputs.Dropdown(["k_lms", "k_heun", "k_euler", "k_euler_a", "k_dpm_2", "k_dpm_2_a", "DDIM", "k_dpm_fast", "k_dpm_adaptive", "k_dpmpp_2m", "k_dpmpp_2s_a", "k_dpmpp_sde"], label="Sampler", default="k_dpmpp_2s_a"), |
|
gr.inputs.Number(label="Steps", default=50) |
|
] |
|
|
|
outputs = gr.outputs.Image(label="Generated Image", type="pil") |
|
|
|
interface = gr.Interface(generate_image, inputs, outputs, title="<a href='https://huggingface.co/spaces/ImagineAI-Real/ImagineAI-Image-Generator'>ImagineAI-Real/ImagineAI Image Generator</a>", |
|
description="Enter a prompt, click Submit and wait a bit for your image.<br/>If taking too long, duplicate the space.<br/>Or use one of the others:<br><li><a href='https://huggingface.co/spaces/ImagineAI-Real/ImagineAI-Image-Generator'>ImagineAI-Image-Generator</a></li><li><a href='https://huggingface.co/spaces/ImagineAI-Real/ImagineAI-Image-Generator2'>ImagineAI-Image-Generator2</a></li><li><a href='https://huggingface.co/spaces/ImagineAI-Real/ImagineAI-Image-Generator3'>ImagineAI-Image-Generator3</a></li><br>Or <a href='https://ufile.io/vwyih7hj'>download our app</a>.<br>If you got an error, just get the app, that works all the time. Sry", |
|
examples=[]) |
|
|
|
interface.launch() |
|
|