csukuangfj commited on
Commit
ad6bdc6
1 Parent(s): 89e21ae

small fixes

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -164,8 +164,14 @@ with demo:
164
  value=language_choices[0],
165
  )
166
 
 
 
 
 
 
 
167
  model_dropdown = gr.Dropdown(
168
- choices=language_to_models[language_choices[0]],
169
  label="Select a model",
170
  value=language_to_models[language_choices[0]][0],
171
  )
 
164
  value=language_choices[0],
165
  )
166
 
167
+ model_choices = []
168
+ for models in language_to_models.values():
169
+ model_choices.extend(models)
170
+
171
+ logging.info(model_choices)
172
+
173
  model_dropdown = gr.Dropdown(
174
+ choices=model_choices,
175
  label="Select a model",
176
  value=language_to_models[language_choices[0]][0],
177
  )