Merge pull request #126 from roboflow/classification-inference-fixes

Classification Inference Fix - PredictionGroup

Author: Tyler-Odenthal

roboflow/__init__.py

@@ -8,7 +8,7 @@
 from roboflow.core.project import Project
 from roboflow.core.workspace import Workspace
 
-__version__ = "0.2.32"
+__version__ = "0.2.33"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/models/classification.py

@@ -47,6 +47,7 @@ def predict(self, image_path, hosted=False):
             # Create buffer
             buffered = io.BytesIO()
             image.save(buffered, quality=90, format="JPEG")
+            img_dims = image.size
             # Base64 encode image
             img_str = base64.b64encode(buffered.getvalue())
             img_str = img_str.decode("ascii")
@@ -66,7 +67,10 @@ def predict(self, image_path, hosted=False):
             raise Exception(resp.text)
 
         return PredictionGroup.create_prediction_group(
-            resp.json(), image_path=image_path, prediction_type=CLASSIFICATION_MODEL
+            resp.json(),
+            image_dims=img_dims,
+            image_path=image_path,
+            prediction_type=CLASSIFICATION_MODEL,
         )
 
     def load_model(self, name, version):