File size: 533 Bytes
f71801e 5a7ebdc f71801e 5a7ebdc f71801e d5098cd f71801e d5098cd f71801e d5098cd f71801e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
# 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!"}
|