tombetthauser commited on
Commit
aa33c7d
β€’
1 Parent(s): 56c8a56

Filtered out betas from dropdown

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -55,6 +55,7 @@ def load_learned_embed_in_clip(learned_embeds_path, text_encoder, tokenizer, tok
55
 
56
 
57
  ahx_model_list = [model for model in models_list if "ahx" in model.modelId]
 
58
 
59
 
60
  for model in ahx_model_list:
@@ -258,7 +259,7 @@ with gr.Blocks(css=".gradio-container {max-width: 650px}") as new_welcome:
258
  <br>
259
  <a href="http://www.astronaut.horse">http://www.astronaut.horse</a>
260
  ''')
261
- dropdown = gr.Dropdown(list(DROPDOWNS), label="choose style...")
262
 
263
  # with gr.Row():
264
  prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
 
55
 
56
 
57
  ahx_model_list = [model for model in models_list if "ahx" in model.modelId]
58
+ ahx_dropdown_list = [model for model in models_list if "ahx-model" in model.modelId]
59
 
60
 
61
  for model in ahx_model_list:
 
259
  <br>
260
  <a href="http://www.astronaut.horse">http://www.astronaut.horse</a>
261
  ''')
262
+ dropdown = gr.Dropdown([dropdown for dropdown in list(DROPDOWNS) if 'ahx-model' in dropdown], label="choose style...")
263
 
264
  # with gr.Row():
265
  prompt = gr.Textbox(label="image prompt...", elem_id="input-text")