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

more tests

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -162,7 +162,7 @@ def create_demo():
162
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
163
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
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
 
@@ -170,7 +170,7 @@ def create_demo():
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)
172
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
173
- return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
174
 
175
  # Define preferred and additional models directly in the function
176
  preferred_models = [
 
162
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
163
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
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
 
 
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)
172
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
173
+ return gr.update(choices=preferred_models, label="Select Model",value=default_model),True,text
174
 
175
  # Define preferred and additional models directly in the function
176
  preferred_models = [