Update app.py

app.py

@@ -7,6 +7,8 @@ from torchvision import transforms
 from models import MAGVITv2, get_mask_schedule, MMadaModelLM
 from training.prompting_utils import UniversalPrompting
 from PIL import Image
+import spaces
+
 
 def image_transform(image, resolution=256, normalize=True):
     image = transforms.Resize(resolution, interpolation=transforms.InterpolationMode.BICUBIC)(image)