Update app.py

app.py

@@ -25,6 +25,13 @@ model_dropdown = gr.inputs.Dropdown(models, label="Select Model")
 input_text = gr.inputs.Textbox(lines=5, label="Input Text")
 output_text = gr.outputs.Textbox(label="Output Text")
 
-interface = gr.Interface(fn=generate_text, inputs=[model_dropdown, input_text], outputs=output_text)
+def change_model(model_name):
+    global interface
+    interface.close()
+    interface = gr.Interface(generate_text, inputs=[model_name, input_text], outputs=output_text)
+    interface.launch()
 
+interface = gr.Interface(generate_text, inputs=[model_dropdown, input_text], outputs=output_text, capture_session=True)
+
+change_model_button = gr.Interface.load("models/stabilityai/stable-diffusion-2", capture_session=True).launch()
 interface.launch()