Spaces:
Sleeping
Sleeping
yash
commited on
Commit
·
fc17416
1
Parent(s):
35082a5
add pokemon_v1.1 model
Browse files
app.py
CHANGED
@@ -10,7 +10,8 @@ import random
|
|
10 |
def set_pipeline(model_id_repo,scheduler):
|
11 |
|
12 |
model_ids_dict = {
|
13 |
-
"pokemon": "yashAI007/pokemon"
|
|
|
14 |
}
|
15 |
model_id = model_id_repo
|
16 |
model_repo = model_ids_dict.get(model_id)
|
@@ -104,7 +105,7 @@ with block as image_gen:
|
|
104 |
negative_prompt = gr.Textbox(placeholder="what you don't want to generate",label="Negative prompt")
|
105 |
run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
|
106 |
with gr.Accordion(label="Advance Options",open=False):
|
107 |
-
model_selection = gr.Dropdown(choices=["pokemon"],value="pokemon",label="Models")
|
108 |
schduler_selection = gr.Dropdown(choices=["DDIM","Euler","Euler a","UniPC","DPM2 Karras","DPM2 a Karras","PNDM","DPM++ 2M Karras","DPM++ 2M SDE Karras"],value="Euler a",label="Scheduler")
|
109 |
guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
|
110 |
num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
|
|
|
10 |
def set_pipeline(model_id_repo,scheduler):
|
11 |
|
12 |
model_ids_dict = {
|
13 |
+
"pokemon": "yashAI007/pokemon",
|
14 |
+
"pokemon_v1.1":"yashAI007/pokemon_v1.1"
|
15 |
}
|
16 |
model_id = model_id_repo
|
17 |
model_repo = model_ids_dict.get(model_id)
|
|
|
105 |
negative_prompt = gr.Textbox(placeholder="what you don't want to generate",label="Negative prompt")
|
106 |
run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
|
107 |
with gr.Accordion(label="Advance Options",open=False):
|
108 |
+
model_selection = gr.Dropdown(choices=["pokemon","pokemon_v1.1"],value="pokemon",label="Models")
|
109 |
schduler_selection = gr.Dropdown(choices=["DDIM","Euler","Euler a","UniPC","DPM2 Karras","DPM2 a Karras","PNDM","DPM++ 2M Karras","DPM++ 2M SDE Karras"],value="Euler a",label="Scheduler")
|
110 |
guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
|
111 |
num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
|