Spaces:
Runtime error
Runtime error
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() |