Update README.md

README.md

@@ -1,7 +1,11 @@
 ---
 license: mit
+language:
+- en
+pipeline_tag: image-to-text
 ---
 This model is to help determine the type of problem a 3D print has.
+The model uses AlexNet CNN Architecture built using PyTorch
 
 The model trained on images of 3D prints as they are printing as well as post printing.
 Training set of images is about ~5GB
@@ -41,5 +45,4 @@ def preProcess(image):
 
     input_image = transform(image).unsqueeze(0)
     return input_image
-```
-
+```