Update app.py

app.py

@@ -1,8 +1,7 @@
 import torch
 from PIL import Image
 import gradio as gr
-import os
-
+import spaces
 from KandiSuperRes import get_SR_pipeline
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
@@ -10,6 +9,7 @@ sr_pipe2x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=2)
 sr_pipe4x = get_SR_pipeline(device=device, fp16=True, flash=True, scale=4)
 
 
+@spaces.GPU()
 def inference(image, size):
     if image is None:
         raise gr.Error("Image not uploaded")