Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import json
 import os
 import zipfile
@@ -17,6 +18,7 @@ from torchvision.transforms import transforms
 from torchvision.transforms import InterpolationMode
 import torchvision.transforms.functional as TF
 from torch.utils.data import Dataset, DataLoader
+from math import ceil
 from typing import Callable
 from functools import partial
 import spaces.config