use zero gpu

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-# import spaces
+import spaces
 import torch
 import torch.nn.functional as F
 from torch.utils.data import DataLoader
@@ -68,7 +68,7 @@ def preprocess_uploaded_image(uploaded_image, image_size):
     return uploaded_image
 
 # ヒートマップの生成関数
-# @spaces.GPU
+@spaces.GPU
 def get_heatmaps(source_num, x_coords, y_coords, uploaded_image):
     if type(uploaded_image) == str:
         uploaded_image = Image.open(uploaded_image)