John6666 commited on
Commit
2417825
Β·
verified Β·
1 Parent(s): 6e041c7

Upload 2 files

Browse files
Files changed (2) hide show
  1. all_models.py +22 -0
  2. app.py +5 -5
all_models.py CHANGED
@@ -895,4 +895,26 @@ models = [
895
  "CompVis/stable-diffusion-v1-3", #207
896
  "CompVis/stable-diffusion-v1-2", #208
897
  "CompVis/stable-diffusion-v1-1", #209
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
898
  ]
 
895
  "CompVis/stable-diffusion-v1-3", #207
896
  "CompVis/stable-diffusion-v1-2", #208
897
  "CompVis/stable-diffusion-v1-1", #209
898
+ ]
899
+
900
+ models = [
901
+ #"runwayml/stable-diffusion-v1-5",
902
+ "nicky007/stable-diffusion-logo-fine-tuned",
903
+ "stablediffusionapi/three-delicacy", #233
904
+ "stablediffusionapi/three-delicacy-wonto", #234
905
+ "naclbit/trinart_stable_diffusion_v2",
906
+ "dallinmackay/Tron-Legacy-diffusion",
907
+ "digiplay/unstableDiffusersYamerMIX_v3",
908
+ "dallinmackay/Van-Gogh-diffusion",
909
+ "ItsJayQz/Valorant_Diffusion",
910
+ "Fictiverse/Stable_Diffusion_VoxelArt_Model", #204
911
+ "wavymulder/wavyfusion",
912
+ "Yntec/HassanRemix",
913
+ "Yntec/Reddit",
914
+ "Yntec/CinematicReality",
915
+ "runwayml/stable-diffusion-v1-5", #555
916
+ "CompVis/stable-diffusion-v1-4", #530
917
+ "CompVis/stable-diffusion-v1-3", #207
918
+ "CompVis/stable-diffusion-v1-2", #208
919
+ "CompVis/stable-diffusion-v1-1", #209
920
  ]
app.py CHANGED
@@ -26,11 +26,11 @@ default_models = models[:num_models]
26
 
27
 
28
  def extend_choices(choices):
29
- return choices + (num_models - len(choices)) * ['NA']
30
 
31
 
32
  def update_imgbox(choices):
33
- choices_plus = extend_choices(choices)
34
  return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
35
 
36
 
@@ -72,9 +72,9 @@ with gr.Blocks() as demo:
72
  #model_choice.change(update_imgbox, model_choice, output)
73
  #model_choice.change(extend_choices, model_choice, current_models)
74
  model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, interactive = True)
75
- model_choice.change(lambda x: x[0:num_models], [model_choice], [model_choice])\
76
- .then(update_imgbox, model_choice, output)\
77
- .then(extend_choices, model_choice, current_models)
78
  with gr.Row():
79
  gr.HTML(
80
  """
 
26
 
27
 
28
  def extend_choices(choices):
29
+ return choices[0:num_models] + (num_models - len(choices[0:num_models])) * ['NA']
30
 
31
 
32
  def update_imgbox(choices):
33
+ choices_plus = extend_choices(choices[0:num_models])
34
  return [gr.Image(None, label = m, visible = (m != 'NA')) for m in choices_plus]
35
 
36
 
 
72
  #model_choice.change(update_imgbox, model_choice, output)
73
  #model_choice.change(extend_choices, model_choice, current_models)
74
  model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 866 available!', value = default_models, interactive = True)
75
+ #model_choice.change(lambda x: x[0:num_models], [model_choice], [model_choice])
76
+ model_choice.change(update_imgbox, model_choice, output)
77
+ model_choice.change(extend_choices, model_choice, current_models)
78
  with gr.Row():
79
  gr.HTML(
80
  """