more tests

app.py

@@ -159,7 +159,7 @@ def create_demo():
                         default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
                         text = f"""**Current Model**: Llama 4 Maverick (free)
                     **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
-                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),True,text
+                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),text
                     
                     is_authorized = profile.username in AUTHORIZED_USER_IDS
                     if is_authorized:
@@ -173,7 +173,7 @@ def create_demo():
                         default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
                         text = f"""**Current Model**: Llama 4 Maverick (free)
                     **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
-                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),True,text
+                        return gr.update(choices=preferred_models, label="Select Model",value=default_model),text
                         
                 # Define preferred and additional models directly in the function
                 preferred_models = [
@@ -223,7 +223,7 @@ def create_demo():
                 demo.load(
                     fn=check_authorization,
                     inputs=None,
-                    outputs=[model_choice,login_button,model_info]
+                    outputs=[model_choice,model_info]
                 )
                 
                 gr.Markdown("### Uploaded Images")