Update app.py

app.py

@@ -1,11 +1,23 @@
 import gradio as gr
-from transformers import AutoModelForImageClassification, AutoFeatureExtractor
+from transformers import AutoModel, AutoFeatureExtractor, AutoConfig
 import torch
 from PIL import Image
 
 # Carga el modelo y el extractor de características
 model_name = "CarPeAs/reconocimiento-facial"
-model = AutoModelForImageClassification.from_pretrained(model_name)
+
+# Intentar cargar la configuración del modelo primero
+config = AutoConfig.from_pretrained(model_name)
+
+# Imprimir la configuración del modelo para verificar su contenido
+print(config)
+
+# Usar la clase correcta basada en la configuración del modelo
+if config.model_type in ['vit', 'beit', 'deit', 'resnet', 'convnext', 'swin']:
+    model = AutoModelForImageClassification.from_pretrained(model_name)
+else:
+    model = AutoModel.from_pretrained(model_name)
+
 feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
 # Función de inferencia