HirCoir commited on
Commit
4cff669
·
verified ·
1 Parent(s): ae62850

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -92,10 +92,20 @@ def convert_to_speech(text, model_name, remove_md=False):
92
 
93
  return None
94
 
 
 
 
 
 
 
 
 
 
95
  @app.route('/')
96
  def index():
97
  tts_models = get_available_models()
98
- return render_template('index.html', tts_models=tts_models)
 
99
 
100
  @app.route('/api/list_ollama_models')
101
  def list_ollama_models():
 
92
 
93
  return None
94
 
95
+ def set_default_models():
96
+ tts_models = get_available_models()
97
+ ollama_models = get_ollama_models()
98
+
99
+ default_tts_model = "RecomendacionesConMiau" if "RecomendacionesConMiau" in tts_models else None
100
+ default_ollama_model = "llama3.2:1b" if "llama3.2:1b" in ollama_models else None
101
+
102
+ return default_tts_model, default_ollama_model
103
+
104
  @app.route('/')
105
  def index():
106
  tts_models = get_available_models()
107
+ default_tts_model, default_ollama_model = set_default_models()
108
+ return render_template('index.html', tts_models=tts_models, default_tts_model=default_tts_model, default_ollama_model=default_ollama_model)
109
 
110
  @app.route('/api/list_ollama_models')
111
  def list_ollama_models():