dlrl2025-docker / app.py
Greg Burlet
initial commit
62139b4
from fastapi import FastAPI, UploadFile, File
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from fastapi.middleware.cors import CORSMiddleware
from inference import predict_hotdog
import io
from PIL import Image
app = FastAPI()
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], # In production, specify your frontend URL
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Mount static files at /static, not at root
app.mount("/static", StaticFiles(directory="static"), name="static")
@app.get("/")
def index() -> FileResponse:
return FileResponse(path="static/index.html", media_type="text/html")
@app.post("/predict")
async def predict_hotdog_endpoint(file: UploadFile = File(...)):
# Read the uploaded file
contents = await file.read()
# Convert to PIL Image
img = Image.open(io.BytesIO(contents))
# Make prediction
predictions = predict_hotdog(img)
return predictions