Kesheratmex commited on
Commit
f51c3e3
·
1 Parent(s): e70689d

Improve show_classes: auto‑load model, handle errors, simplify output

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -161,15 +161,17 @@ def infer_media(media_path, conf=0.25, iou=0.45, out_res="720p", preset="default
161
 
162
 
163
  def show_classes():
164
- """Devuelve las clases que el modelo conoce (o mensaje si no está cargado)."""
165
- if model is None:
 
 
 
 
 
 
 
 
166
  return "Model not loaded"
167
- names = model.names
168
- if isinstance(names, dict):
169
- class_list = [names[k] for k in sorted(names)]
170
- else:
171
- class_list = names
172
- return ", ".join(class_list)
173
 
174
  # Funciones auxiliares para extraer el recurso de salida desde el dict
175
 
 
161
 
162
 
163
  def show_classes():
164
+ """Devuelve las clases que el modelo conoce. Carga el modelo si no está listo."""
165
+ try:
166
+ m = get_model()
167
+ names = m.names
168
+ if isinstance(names, dict):
169
+ class_list = [names[k] for k in sorted(names)]
170
+ else:
171
+ class_list = names
172
+ return ", ".join(class_list)
173
+ except Exception:
174
  return "Model not loaded"
 
 
 
 
 
 
175
 
176
  # Funciones auxiliares para extraer el recurso de salida desde el dict
177