File size: 2,544 Bytes
2eb58d1 199e379 2eb58d1 046a2a3 2eb58d1 11ca63e 2eb58d1 a17b7c9 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 |
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(text,steps,width,height,images,diversity):
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()
|