Chris Addis commited on
Commit
93054df
·
1 Parent(s): 8cfc55d
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -177,7 +177,7 @@ def create_demo():
177
  # Calculate all models once
178
  all_models_list = preferred_models + additional_models
179
 
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
@@ -210,7 +210,8 @@ def create_demo():
210
  label="Content Type",
211
  value="Museum Object"
212
  )
213
-
 
214
  model_info = gr.Markdown("",
215
  elem_id="model-info-display"
216
  )
 
177
  # Calculate all models once
178
  all_models_list = preferred_models + additional_models
179
 
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
 
210
  label="Content Type",
211
  value="Museum Object"
212
  )
213
+
214
+ #markdown for current model costings
215
  model_info = gr.Markdown("",
216
  elem_id="model-info-display"
217
  )