app.py

# app.py
import os
from fastapi import FastAPI
from transformers import AutoModelForImageClassification

# Crea la carpeta de caché si no existe
cache_dir = "/tmp/cache"
os.makedirs(cache_dir, exist_ok=True)

# Inicializa la aplicación
app = FastAPI()

# Carga el modelo especificando el directorio de caché
model_name = "nombre_del_modelo"
model = AutoModelForImageClassification.from_pretrained(model_name, cache_dir=cache_dir)

@app.get("/")
async def root():
    return {"message": "¡Aplicación funcionando correctamente!"}