minoruskore commited on
Commit
2a9e07e
·
1 Parent(s): 237774d

Actualizar validación de archivos en la interfaz de evaluación para aceptar extensiones .safetensor y .safetensors

Browse files
Files changed (1) hide show
  1. evaluation.py +4 -2
evaluation.py CHANGED
@@ -35,8 +35,10 @@ def evaluate_interface(model_file, model_type, num_clases, test_dataloader):
35
  return "Por favor, carga un archivo .safetensor"
36
 
37
  # Verificamos que el archivo sea .safetensor
38
- if not model_file.name.endswith(".safetensor"):
39
- return "Por favor, carga un archivo con extensión .safetensor"
 
 
40
 
41
  # Evaluamos el modelo
42
  accuracy = cargar_evaluar_modelo(
 
35
  return "Por favor, carga un archivo .safetensor"
36
 
37
  # Verificamos que el archivo sea .safetensor
38
+ if not model_file.name.endswith(".safetensor") or model_file.name.endswith(
39
+ ".safetensors"
40
+ ):
41
+ return "Por favor, carga un archivo con extensión .safetensor o .safetensors"
42
 
43
  # Evaluamos el modelo
44
  accuracy = cargar_evaluar_modelo(