Chris Addis commited on
Commit
7938411
·
1 Parent(s): 93054df
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -180,7 +180,7 @@ def create_demo():
180
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
181
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
182
 
183
- return gr.Dropdown(choices=preferred_models, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
184
  else:
185
  all_models_list = preferred_models
186
 
@@ -188,7 +188,7 @@ def create_demo():
188
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
189
  text = f"""**Current Model**: Llama 4 Maverick (free)
190
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
191
- return gr.Dropdown(choices=preferred_models, label="Select Model",value=default_model),True,text
192
 
193
  length_choice = gr.Radio(
194
  choices=["short", "medium", "long"],
 
180
  text = f"""**Current Model**: Gemini 2.0 Flash (cheap)
181
  **Estimated cost per 100 Images**: {MODEL_PRICING["google/gemini-2.0-flash-001"]}"""
182
 
183
+ return gr.Update(choices=preferred_models, label="Select Model",value="google/gemini-2.0-flash-001"),False,text
184
  else:
185
  all_models_list = preferred_models
186
 
 
188
  default_model = "meta-llama/llama-4-maverick:free"#preferred_models[0][1] # get free model
189
  text = f"""**Current Model**: Llama 4 Maverick (free)
190
  **Estimated cost per 100 Images**: {MODEL_PRICING["meta-llama/llama-4-maverick:free"]}"""
191
+ return gr.Update(choices=preferred_models, label="Select Model",value=default_model),True,text
192
 
193
  length_choice = gr.Radio(
194
  choices=["short", "medium", "long"],