Uploading Trashify box detection model app.py

app.py

@@ -13,9 +13,14 @@ from transformers import AutoModelForObjectDetection
 # Note: Can load from Hugging Face or can load from local 
 model_save_path = "mrdbourke/rt_detrv2_finetuned_trashify_box_detector_v1"
 
-# Load the model and preprocessor
-# Because this app.py file is running directly on Hugging Face Spaces, the model will be loaded from the Hugging Face Hub
+# Note: Because this app.py file is running directly on Hugging Face Spaces, the processor/model will be loaded from the Hugging Face Hub
+# Load the processor
 image_processor = AutoImageProcessor.from_pretrained(model_save_path)
+# Default to resizing images to 640x640, makes things simple as well as 
+# helps handle edge case images with strange dimensions
+image_processor.size = {"height": 640, 
+                            "width": 640}
+# Load the model
 model = AutoModelForObjectDetection.from_pretrained(model_save_path)
 
 # Set the target device (use CUDA/GPU if it is available)