yash commited on
Commit
fc17416
·
1 Parent(s): 35082a5

add pokemon_v1.1 model

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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)