Update README.md

README.md

@@ -18,3 +18,28 @@ Future updates will include
 - Determine if the image is a 3D print or not
 - Determine if the image is during printing or once complete
 
+
+To make an inference
+
+Classes
+```
+class_names = {0: 'good', 1: 'spaghetti', 2: 'stringing', 3: 'underextrusion'}
+```
+
+Pre-Process the image using the following python function
+```
+def preProcess(image):
+    # Open the image from raw bytes
+    image = Image.open(BytesIO(image)).convert('RGB')
+
+    transform = transforms.Compose([
+        transforms.Resize(227),
+        transforms.CenterCrop(227),
+        transforms.ToTensor(),
+        transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))
+    ])
+
+    input_image = transform(image).unsqueeze(0)
+    return input_image
+```
+