Add styles dropdown

app.py

@@ -14,8 +14,15 @@ my_token = api_key
 with init_empty_weights():
   pipe = pipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=my_token).to("cuda")
 
-def image_prompt(thing):
-  prompt = "A children's drawing of a " + thing
+DROPDOWNS = {
+    "gustav": " by dan mumford and gustav klimt and john harris and jean delville and victo ngai and josan gonzalez",
+    "hiyao": " by studio ghibli, by miyazaki, watercolor painting",
+    "vinny": " painting by Vincent van Gogh",
+    "kids": " drawn by a child",
+}
+
+def image_prompt(prompt, dropdown):
+  prompt = prompt + DROPDOWNS[dropdown]
   return pipe(prompt=prompt, height=512, width=664).images[0]
 
 with gradio.Blocks(css="""
@@ -89,9 +96,10 @@ with gradio.Blocks(css="""
     border-radius: 0;
   }
 """) as demo:
-  animal = gradio.Textbox(label="generate a children's drawing of a...", elem_id="input-text")
+  dropdown = gradio.Dropdown(["danny", "gustav", "hiyao", "vinny"])
+  prompt = gradio.Textbox(label="image prompt...", elem_id="input-text")
   output = gradio.Image(elem_id="output-image")
   go_button = gradio.Button("draw it!", elem_id="go-button")
-  go_button.click(fn=image_prompt, inputs=animal, outputs=output)
+  go_button.click(fn=image_prompt, inputs=[prompt, dropdown], outputs=output)
 
 demo.launch()