File size: 1,986 Bytes
2a75d54
 
 
 
 
 
 
 
7b95f1c
2a75d54
 
 
 
 
 
 
 
 
 
 
7b95f1c
2a75d54
 
 
 
 
 
 
 
 
ea80591
 
2a75d54
 
7d33e57
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
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(model="Dreamlike Diffusion", prompt="", seed=0, negative_prompt="", sampler="k_dpmpp_2s_a", steps=50):
    data = "?model=" + model + "&prompt=" + prompt + "&seed=" + str(seed) + "&negative_prompt=" + negative_prompt + "&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.Dropdown(['Deliberate', 'Dreamlike Diffusion', 'Dreamshaper', 'Elden Ring Diffusion', 'Epic Diffusion', 'Experience',  'FaeTastic', 'Inkpunk Diffusion', 'Kenshi', 'Mega Merge Diffusion', 'Midjourney Diffusion',  'Midjourney PaintArt',  'ModernArt Diffusion', 'Movie Diffusion',  'NeverEnding Dream', 'PRMJ', 'ProtoGen', 'RealBiter',  'RCNZ Gorilla With A Brick',  'RPG', 'Seek.art MEGA', 'Samdoesarts Ultmerge',  'Seek.art MEGA', 'Unstable Ink Dream', 'Van Gogh Diffusion',  'VinteProtogenMix' ], label="Model", default="Dreamlike Diffusion"),
    gr.inputs.Textbox(label="Prompt"),
    gr.inputs.Number(label="Seed", default=0),
    gr.inputs.Textbox(label="Negative Prompt", default=""),
    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="", 
                         description="<center></center>", 
                         examples=[])

interface.launch()