Upload 4 files

Dockerfile

@@ -0,0 +1,19 @@
+# Dockerfile
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system deps
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    build-essential \
+    libsndfile1 \
+    && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+EXPOSE 7860
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]

complete_model_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b2c1ab8e0fb0839b2447c57d6b9c3ae219a052bc3aceaafc3b176e899dcfe56f
+size 27908672

main.py

@@ -0,0 +1,82 @@
+import os
+import io
+import logging
+from typing import Tuple
+
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from PIL import Image
+import numpy as np
+import tensorflow as tf
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger("vehicle-predictor")
+
+MODEL_FILENAME = "complete_model_model.h5"
+MODEL_PATH = os.path.join(os.path.dirname(__file__), MODEL_FILENAME)
+IMG_SIZE = (224, 224)
+
+CLASS_NAMES = [
+    'Ambulance', 'Bicycle', 'Boat', 'Bus', 'Car', 'Helicopter', 'Limousine',
+    'Motorcycle', 'PickUp', 'Segway', 'Snowmobile', 'Tank', 'Taxi', 'Truck', 'Van'
+]
+
+app = FastAPI(title="Vehicle Type Predictor")
+
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # you can tighten this later if needed
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# Load model at startup
+try:
+    logger.info("🚀 Loading model...")
+    model = tf.keras.models.load_model(MODEL_PATH)
+    logger.info("✅ Model loaded successfully.")
+except Exception as e:
+    logger.exception("❌ Model load failed")
+    model = None
+
+
+class PredictionResponse(BaseModel):
+    label: str
+    confidence: float
+
+
+def preprocess_image_file(file_bytes: bytes) -> np.ndarray:
+    img = Image.open(io.BytesIO(file_bytes)).convert("RGB")
+    img = img.resize(IMG_SIZE)
+    arr = np.asarray(img).astype("float32") / 255.0
+    arr = np.expand_dims(arr, axis=0)
+    return arr
+
+
+@app.post("/predict", response_model=PredictionResponse)
+async def predict(file: UploadFile = File(...)):
+    if model is None:
+        raise HTTPException(status_code=503, detail="Model not loaded")
+
+    if not file.content_type.startswith("image/"):
+        raise HTTPException(status_code=400, detail="File must be an image")
+
+    try:
+        contents = await file.read()
+        x = preprocess_image_file(contents)
+        preds = model.predict(x)
+        idx = int(np.argmax(preds[0]))
+        label = CLASS_NAMES[idx]
+        confidence = float(preds[0][idx])
+        logger.info(f"Predicted {label} ({confidence:.4f}) for {file.filename}")
+        return PredictionResponse(label=label, confidence=confidence)
+    except Exception as e:
+        logger.exception("Prediction failed")
+        raise HTTPException(status_code=500, detail="Prediction failed")
+
+
+@app.get("/health")
+def health():
+    return {"status": "ok", "model_loaded": model is not None}

requirements.txt

@@ -0,0 +1,56 @@
+absl-py==2.3.1
+annotated-types==0.7.0
+anyio==4.11.0
+astunparse==1.6.3
+certifi==2025.10.5
+charset-normalizer==3.4.4
+click==8.3.0
+fastapi==0.119.0
+flatbuffers==25.9.23
+gast==0.6.0
+google-pasta==0.2.0
+grpcio==1.75.1
+h11==0.16.0
+h5py==3.15.0
+httptools==0.7.1
+idna==3.11
+keras==3.11.3
+libclang==18.1.1
+Markdown==3.9
+markdown-it-py==4.0.0
+MarkupSafe==3.0.3
+mdurl==0.1.2
+ml_dtypes==0.5.3
+namex==0.1.0
+numpy==2.3.3
+opt_einsum==3.4.0
+optree==0.17.0
+packaging==25.0
+pillow==11.3.0
+protobuf==6.32.1
+pydantic==2.12.2
+pydantic_core==2.41.4
+Pygments==2.19.2
+python-dotenv==1.1.1
+python-multipart==0.0.20
+PyYAML==6.0.3
+requests==2.32.5
+rich==14.2.0
+setuptools==80.9.0
+six==1.17.0
+sniffio==1.3.1
+starlette==0.48.0
+tensorboard==2.20.0
+tensorboard-data-server==0.7.2
+tensorflow==2.20.0
+termcolor==3.1.0
+typing-inspection==0.4.2
+typing_extensions==4.15.0
+urllib3==2.5.0
+uvicorn==0.37.0
+uvloop==0.21.0
+watchfiles==1.1.1
+websockets==15.0.1
+Werkzeug==3.1.3
+wheel==0.45.1
+wrapt==1.17.3