Update app.py

app.py

@@ -18,6 +18,8 @@ def resize_image(image):
     image = image.resize(model_input_size, Image.BILINEAR)
     return image
 
+@spaces.GPU
+@torch.inference_mode()
 def inference(image):
     # Check for CUDA and set the device inside inference
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")