Update app.py

app.py

@@ -53,8 +53,10 @@ def load_fn(models):
 load_fn(models)
 
 tags_plus_models_to_list=[]
+list_tags=[]
 for tag_plus_m in tags_plus_models:
     tags_plus_models_to_list.append([tag_plus_m[0],tag_plus_m[2]])
+    list_tags.append(tag_plus_m[0])
     
 def test_pass(test):
     if test==os.getenv('p'):
@@ -309,9 +311,12 @@ def make_me():
             with gr.Row():
                 nb_images_by_prompt=gr.Number(2,label="Number of images by prompt:",interactive=True)
                 nb_of_models_to_gen=gr.Number(10,label="Number of Models:",interactive=True)
-                list_models_in_tag=gr.Dropdown(label="Tag",choices=list(tags_plus_models_to_list),value=tags_plus_models_to_list[1][1])
-                index_first_model=gr.Dropdown(label="First model",choices=tags_plus_models_to_list[1][1], type="index")
-                list_models_in_tag.change(lambda x:gr.Dropdown(choices=x),list_models_in_tag,index_first_model)
+                #list_models_in_tag=gr.Dropdown(label="Tag",choices=list(tags_plus_models_to_list),value=tags_plus_models_to_list[1][1])
+                #index_first_model=gr.Dropdown(label="First model",choices=tags_plus_models_to_list[1][1], type="index")
+                #list_models_in_tag.change(lambda x:gr.Dropdown(choices=x),list_models_in_tag,index_first_model)
+                index_tag=gr.Dropdown(label="Tag",choices=list(list_tags),type="index")
+                index_first_model=gr.Dropdown(label="First model",choices=list(tags_plus_models[1][2]), type="index")
+                index_tag.change(lambda x:gr.Dropdown(choices=list(tags_plus_models[index_tag][2])),index_tag,index_first_model)
             with gr.Row():
                 disp_info=gr.Textbox(label="Info")
                 load_info=gr.Button("Load info")