Add model warmup

main.py

@@ -33,6 +33,14 @@ EXAMPLE_IMAGES = [
     ['examples/P_00001_LEFT_cc.jpg', 'examples/P_00001_LEFT_mlo.jpeg'],
 ]
 
+# Model warmup
+test_images = np.random.randint(0, 255, (2, INPUT_HEIGHT, INPUT_WIDTH))
+test_images = torch.from_numpy(test_images).to(device)
+test_images = test_images.unsqueeze(0)  # Add batch dimension
+for _ in range(10):
+    _, _, _ = BILATERAL_MODEL(test_images)
+test_images = None
+
 
 def filter_files(files: List) -> List:
     """Filter uploaded files.