Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
|
|
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():
|