ViannyCruz commited on
Commit
da48ebe
·
verified ·
1 Parent(s): d94c02e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -7
app.py CHANGED
@@ -104,11 +104,13 @@ def admin_required(f):
104
  # ------------------------------------------------------------------ #
105
  def load_model():
106
  global model
107
- model_files = [f for f in os.listdir('.') if f.endswith('.h5')]
 
 
108
  if not model_files:
109
- print("ERROR: No hay archivos .h5 en el directorio")
110
  return False
111
- model_path = model_files[0]
112
  print(f"Cargando modelo: {model_path}")
113
  try:
114
  from tensorflow.keras.applications import EfficientNetB0
@@ -727,11 +729,16 @@ def _load_model_background():
727
  global _model_error
728
  print("Iniciando carga de modelo en background...")
729
  try:
730
- load_model()
731
- print("Modelo cargado exitosamente en background")
 
 
 
 
732
  except Exception as e:
733
- _model_error = str(e)
734
- print(f"Error cargando modelo en background: {e}")
 
735
 
736
  _model_thread = threading.Thread(target=_load_model_background, daemon=True)
737
  _model_thread.start()
 
104
  # ------------------------------------------------------------------ #
105
  def load_model():
106
  global model
107
+ # Usar ruta absoluta para evitar problemas con directorio de trabajo
108
+ app_dir = os.path.dirname(os.path.abspath(__file__))
109
+ model_files = [f for f in os.listdir(app_dir) if f.endswith('.h5')]
110
  if not model_files:
111
+ print(f"ERROR: No hay archivos .h5 en {app_dir}")
112
  return False
113
+ model_path = os.path.join(app_dir, model_files[0])
114
  print(f"Cargando modelo: {model_path}")
115
  try:
116
  from tensorflow.keras.applications import EfficientNetB0
 
729
  global _model_error
730
  print("Iniciando carga de modelo en background...")
731
  try:
732
+ result = load_model()
733
+ if result:
734
+ print("Modelo cargado exitosamente en background")
735
+ else:
736
+ _model_error = "load_model() retornó False — ver logs del servidor"
737
+ print(_model_error)
738
  except Exception as e:
739
+ import traceback
740
+ _model_error = traceback.format_exc()
741
+ print(f"Error cargando modelo en background: {_model_error}")
742
 
743
  _model_thread = threading.Thread(target=_load_model_background, daemon=True)
744
  _model_thread.start()