File size: 2,580 Bytes
2eb58d1 199e379 2eb58d1 046a2a3 2eb58d1 11ca63e 2eb58d1 c21f330 2eb58d1 c21f330 2eb58d1 199e379 2eb58d1 199e379 2eb58d1 c21f330 2eb58d1 c21f330 2eb58d1 c21f330 2eb58d1 c21f330 2eb58d1 c21f330 2eb58d1 11ca63e 2eb58d1 c21f330 2eb58d1 |
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 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
from turtle import width
import gradio as gr
import random
import inspect
latent = gr.Interface.load("spaces/multimodalart/latentdiffusion")
print(latent)
rudalle = gr.Interface.load("spaces/multimodalart/rudalle")
print(rudalle)
guided = gr.Interface.load("spaces/EleutherAI/clip-guided-diffusion")
print(guided)
def text2image_latent(which_tab,text,steps,width,height,images,diversity):
print(which_tab)
image = latent(text, steps, width, height, images, diversity)[0]
return(image)
def text2image_rudalle(text,aspect,model):
image = rudalle(text,aspect,model)[0]
print(image)
return(image)
def text2image_guided(text):
image = guided(text,'',10,600,0,0,0,random.randint(0,2147483647),'',50,32)
print(image)
image = image[0]
return(image)
block = gr.Blocks()
with block:
text = gr.inputs.Textbox(placeholder="Try writing something..")
which_tab = "Latent Diffusion"
with gr.Tab("latent"):
which_tab = "Latent Diffusion"
steps = gr.inputs.Slider(label="Steps - more steps can increase quality but will take longer to generate",default=45,maximum=50,minimum=1,step=1)
width = gr.inputs.Radio(label="Width", choices=[32,64,128,256],default=256)
height = gr.inputs.Radio(label="Height", choices=[32,64,128,256],default=256)
images = gr.inputs.Slider(label="Images - How many images you wish to generate", default=2, step=1, minimum=1, maximum=4)
diversity = gr.inputs.Slider(label="Diversity scale - How different from one another you wish the images to be",default=5.0, minimum=1.0, maximum=15.0)
get_image_latent = gr.Button("Generate Image")
with gr.Tab("rudalle"):
which_tab = "ruDALLE"
aspect = gr.inputs.Radio(label="Aspect Ratio", choices=["Square", "Horizontal", "Vertical"],default="Square")
model = gr.inputs.Dropdown(label="Model", choices=["Surrealism","Realism", "Emoji"], default="Surrealism")
get_image_rudalle = gr.Button("Generate Image")
with gr.Tab("guided"):
which_tab = "Guided Diffusion"
get_image_guided = gr.Button("Generate Image")
get_image = gr.Button("Generate Image")
with gr.Column():
with gr.Row():
image = gr.outputs.Image()
print(which_tab)
get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=image)
get_image_rudalle.click(text2image_rudalle, inputs=[text,aspect,model], outputs=image)
get_image_guided.click(text2image_guided, inputs=text, outputs=image)
block.launch()
|