Update app.py

app.py

@@ -97,6 +97,7 @@ def detect(image):
         }
 
 def classify(image):
+    image_width, image_height = image.size
     detection = detect(image)
     
     if detection["top"] == 0 and detection["left"] == 0 and detection["width"] == 0 and detection["height"] == 0:
@@ -123,7 +124,9 @@ def classify(image):
     print(f"{formatted_time} {sorted_result}")
     return {
         "detection": detection,
-        "classification": sorted_result
+        "classification": sorted_result,
+        "image_width": image_width,
+        "image_height": image_height
     }