Update app.py

app.py

@@ -29,6 +29,11 @@ def generate(prompts):
   images = generator(list(prompts)).images
   return [images]
 
+def set_model(current_model_index):
+    global current_model
+    current_model = models[current_model_index]
+    return gr.update(value=f"{current_model['name']}")
+
 with gr.Blocks() as demo:
     gr.HTML(