Spaces:
Sleeping
Sleeping
Kesheratmex
commited on
Commit
·
f51c3e3
1
Parent(s):
e70689d
Improve show_classes: auto‑load model, handle errors, simplify output
Browse files
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
|
| 165 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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 |
|