add gpu annotation

app.py

@@ -4,12 +4,14 @@ import sys
 
 import gradio as gr
 import numpy as np
+import spaces
 
-# import spaces
 # from huggingface_hub import hf_hub_download
 from huggingface_hub import snapshot_download
 from PIL import Image, ImageDraw, ImageFont
 
+zero = torch.Tensor([0]).cuda()
+
 # Set the working directory to the root directory
 # root_dir = os.path.abspath("..")
 # os.chdir(root_dir)
@@ -137,6 +139,7 @@ def viz(x, cmap="gray", vmin=0, vmax=1):
     return img_array
 
 
+@spaces.GPU
 def forward(model, idx, rate):
     if rate == 4:
         dataset = val_dataset_4x