apolinario
commited on
Commit
•
a0c074d
1
Parent(s):
e94d44b
tentative use of new api
Browse files
app.py
CHANGED
@@ -29,25 +29,25 @@ def text2image_guided(text):
|
|
29 |
|
30 |
block = gr.Blocks()
|
31 |
|
32 |
-
with
|
33 |
text = gr.inputs.Textbox(placeholder="Try writing something..")
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
#get_image = gr.Button("Generate Image")
|
52 |
with gr.Column():
|
53 |
with gr.Row():
|
@@ -57,4 +57,4 @@ with block:
|
|
57 |
get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=image)
|
58 |
get_image_rudalle.click(text2image_rudalle, inputs=[text,aspect,model], outputs=image)
|
59 |
get_image_guided.click(text2image_guided, inputs=text, outputs=image)
|
60 |
-
|
|
|
29 |
|
30 |
block = gr.Blocks()
|
31 |
|
32 |
+
with gr.Blocks() as mindseye:
|
33 |
text = gr.inputs.Textbox(placeholder="Try writing something..")
|
34 |
+
with gr.Tabs():
|
35 |
+
with gr.TabItem("Latent"):
|
36 |
+
which_tab = "Latent Diffusion"
|
37 |
+
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)
|
38 |
+
width = gr.inputs.Radio(label="Width", choices=[32,64,128,256],default=256)
|
39 |
+
height = gr.inputs.Radio(label="Height", choices=[32,64,128,256],default=256)
|
40 |
+
images = gr.inputs.Slider(label="Images - How many images you wish to generate", default=2, step=1, minimum=1, maximum=4)
|
41 |
+
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)
|
42 |
+
get_image_latent = gr.Button("Generate Image")
|
43 |
+
with gr.TabItem("Rudalle"):
|
44 |
+
which_tab = "ruDALLE"
|
45 |
+
aspect = gr.inputs.Radio(label="Aspect Ratio", choices=["Square", "Horizontal", "Vertical"],default="Square")
|
46 |
+
model = gr.inputs.Dropdown(label="Model", choices=["Surrealism","Realism", "Emoji"], default="Surrealism")
|
47 |
+
get_image_rudalle = gr.Button("Generate Image")
|
48 |
+
with gr.TabItem("guided"):
|
49 |
+
which_tab = "Guided Diffusion"
|
50 |
+
get_image_guided = gr.Button("Generate Image")
|
51 |
#get_image = gr.Button("Generate Image")
|
52 |
with gr.Column():
|
53 |
with gr.Row():
|
|
|
57 |
get_image_latent.click(text2image_latent, inputs=[text,steps,width,height,images,diversity], outputs=image)
|
58 |
get_image_rudalle.click(text2image_rudalle, inputs=[text,aspect,model], outputs=image)
|
59 |
get_image_guided.click(text2image_guided, inputs=text, outputs=image)
|
60 |
+
mindseye.launch()
|