Add: README.md documentation

Author: Baichuan Sun

.gitignore

@@ -138,3 +138,4 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
+.DS_Store

README.md

@@ -34,7 +34,6 @@ def initialize(self, ctx):
  else "cpu"
  )
  model_dir = properties.get("model_dir")
- print(model_dir)
 
  manifest = ctx.manifest
  logger.error(manifest)
@@ -64,7 +63,7 @@ def preprocess(self, data):
  if image is None:
  image = data[0].get("body")
 
- chipstack_transform = transforms.Compose(
+ image_transform = transforms.Compose(
  [
  # must be consistent with model training
  transforms.Resize((96, 128)),
@@ -78,7 +77,7 @@ def preprocess(self, data):
  image = Image.open(io.BytesIO(image)).convert(
  "RGB"
  ) # in case of an alpha channel
- image = chipstack_transform(image).unsqueeze_(0)
+ image = image_transform(image).unsqueeze_(0)
  return image
 
  def inference(self, img):