app.py

from transformers import ViTFeatureExtractor, AutoModelForImageClassification

def classify_image(image):
    try:
        feature_extractor = ViTFeatureExtractor.from_pretrained("ismaeltorres00/ModeloFinalEuroSat")
    except OSError as e:
        # Manejo del error si el archivo no existe
        print("No se pudo encontrar el archivo preprocessor_config.json. Verifica el repositorio.")
        raise e
    
    model = AutoModelForImageClassification.from_pretrained("ismaeltorres00/ModeloFinalEuroSat")
    inputs = feature_extractor(images=image, return_tensors="pt")
    outputs = model(**inputs)
    logits = outputs.logits

    # Aquí procesarías los logits para obtener la clasificación
    return logits.argmax(-1).item()