Chris Addis commited on
Commit
3830538
·
1 Parent(s): 2cfe564

more tests

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -137,6 +137,12 @@ def create_demo():
137
  ##########################################################################
138
  # Function to check authorization
139
  def check_authorization(profile: gr.OAuthProfile | None):
 
 
 
 
 
 
140
 
141
  is_authorized = profile.username in AUTHORIZED_USER_IDS
142
  if is_authorized:
@@ -164,8 +170,6 @@ def create_demo():
164
 
165
  return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
166
  else:
167
- all_models_list = preferred_models
168
-
169
  # Default model value
170
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
171
  text = f"""**Current Model**: Llama 4 Maverick (free)
 
137
  ##########################################################################
138
  # Function to check authorization
139
  def check_authorization(profile: gr.OAuthProfile | None):
140
+ if profile is None:
141
+ # Default model value
142
+ default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
143
+ text = f"""**Current Model**: Llama 4 Maverick (free)
144
+ **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
145
+ return gr.update(choices=preferred_models, label="Select Model",value=default_model),True,text
146
 
147
  is_authorized = profile.username in AUTHORIZED_USER_IDS
148
  if is_authorized:
 
170
 
171
  return gr.update(choices=preferred_models_auth, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
172
  else:
 
 
173
  # Default model value
174
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
175
  text = f"""**Current Model**: Llama 4 Maverick (free)