OAuth

app.py

@@ -170,7 +170,7 @@ def create_demo():
 
                 #demo.load(check_authorization, inputs=None, outputs=[auth_state])
 
-                if check_authorization(None):
+                with gr.Group(visible=False) as unauthenticated_ui:
                     # free version - yeah baby
                     login_button = gr.LoginButton()
 
@@ -183,7 +183,11 @@ def create_demo():
         
                     # Default model value
                     default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
-
+                demo.load(
+                    fn=check_authorization,
+                    inputs=None,
+                    outputs=[unauthenticated_ui]
+                )
                 ################################## normal stuff below  
                 upload_button = gr.UploadButton(
                     "Click to Upload Images",