Initial age classifier API

app/main.py

@@ -0,0 +1,21 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from transformers import pipeline
+from PIL import Image
+import io
+
+app = FastAPI()
+
+# Load the model once when the server starts
+pipe = pipeline("image-classification", model="nateraw/vit-age-classifier")
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    # Read image file
+    image_data = await file.read()
+    image = Image.open(io.BytesIO(image_data)).convert("RGB")
+
+    # Run prediction
+    results = pipe(image)
+    
+    return JSONResponse(content={"results": results})

requirements.txt

@@ -0,0 +1,5 @@
+transformers
+torch
+fastapi
+uvicorn
+pillow