Update app.py

app.py

@@ -5,6 +5,7 @@ import cv2 # Using OpenCV for image loading/processing
 import albumentations as A
 from albumentations.pytorch import ToTensorV2
 import gradio as gr
+import spaces
 
 import segmentation_models_pytorch as smp
 from train_unet import UNetLitModule # Import the Lightning Module definition