edit login button

app.py

@@ -226,6 +226,12 @@ def create_demo():
                     outputs=[model_choice,model_info]
                 )
                 
+                login_button.login(
+                    fn=check_authorization,
+                    inputs=None, # The user profile is automatically passed on login
+                    outputs=[model_choice, model_info]
+                )
+
                 gr.Markdown("### Uploaded Images")
                 input_gallery = gr.Gallery(
                     label="Uploaded Image Previews", columns=3, height=150,