Update

app.py

@@ -8,6 +8,7 @@ import sys
 import gradio as gr
 import huggingface_hub
 import PIL.Image
+import spaces
 import torch
 import torch.nn as nn
 
@@ -40,6 +41,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 model = load_model(device)
 
 
+@spaces.GPU
 @torch.inference_mode()
 def run(image_file: str, load_size: int = 512) -> PIL.Image.Image:
     tensor, orig_size = read_img_path(image_file, load_size)