Update app.py

app.py

@@ -13,6 +13,7 @@ def inference(input_img, conf_threshold, iou_threshold):
     model.conf = conf_threshold
     model.iou = iou_threshold
     # Perform inference
+    image_size = input_img.shape
     results = model(input_img, size=image_size)
 
     # Optionally, show detection bounding boxes on image