Skip to content

sofieff/MLOps-DSR-batch-43

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

9 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿš€ MLOps Class: Image Classifier Deployment on GCP

This applied class project is the final MLOps deployment from the Data Science Retreat program. It demonstrates a robust, traceable pipeline for a ResNet Image Classifier (Fresh/Rotten Fruit classification).

โš ๏ธ Deployment Status: The live application was successfully deployed on Google Cloud Platform (GCP) but has been deleted to manage costs. This repository maintains the complete, documented infrastructure for demonstrating pipeline and deployment expertise.

โš™๏ธ Architecture and Key Technologies

Component Technology Purpose
Serving Endpoint FastAPI Provides a scalable, high-performance API for image prediction.
Data Validation Pydantic Ensures robust API request and response data integrity.
Artifact Management Weights & Biases (W&B) Dynamically fetches version-controlled model weights (best_model.pth) via the W&B API at container startup.
Model PyTorch / ResNet Optimized model loading and inference using torch.inference_mode().
Containerization Docker Ensures a portable, consistent runtime environment for deployment.
Cloud Target Google Cloud Platform (GCP) Repository showcases configuration for services like Cloud Run or Compute Engine.

๐Ÿ“ Repository Structure

The core application logic is contained within the app/ directory, following industry standards for containerized microservices.

  • app/main.py: Defines the FastAPI endpoint (/predict).
  • app/model.py: Crucial: Handles artifact fetching, environment configuration, and PyTorch model loading from W&B.
  • Dockerfile: Defines the build environment and startup command.
  • requirements.txt: Specifies all dependencies, including wandb and torch.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published