Spaces:
Runtime error
Runtime error
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)
|