caddis commited on
Commit
3e43e85
·
1 Parent(s): 7b3d9d6

update message

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -160,7 +160,7 @@ def create_demo():
160
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
161
  text = f"""**Current Model**: Llama 4 Maverick (free)
162
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
163
- return gr.update(choices=preferred_models, label="Select Model",value=default_model),text,""
164
 
165
  is_authorized = profile.username in AUTHORIZED_USER_IDS
166
  if is_authorized:
@@ -168,7 +168,7 @@ def create_demo():
168
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
169
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
170
 
171
- return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),text
172
  else:
173
  # Default model value
174
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
 
160
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
161
  text = f"""**Current Model**: Llama 4 Maverick (free)
162
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
163
+ return gr.update(choices=preferred_models, label="Select Model",value=default_model),text,f"Logged in as: {profile.username}"
164
 
165
  is_authorized = profile.username in AUTHORIZED_USER_IDS
166
  if is_authorized:
 
168
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
169
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
170
 
171
+ return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),text,f"Logged in as: {profile.username}"
172
  else:
173
  # Default model value
174
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model