Update app.py

app.py

@@ -1,11 +1,11 @@
 import os
-
 import cv2
 import gradio as gr
 import torch
 from basicsr.archs.srvgg_arch import SRVGGNetCompact
 from gfpgan.utils import GFPGANer
 from realesrgan.utils import RealESRGANer
+import spaces
 
 os.system("pip freeze")
 # download weights
@@ -45,6 +45,7 @@ os.makedirs('output', exist_ok=True)
 
 
 # def inference(img, version, scale, weight):
+@spaces.GPU(duration=90)
 def inference(img, version, scale):
     # weight /= 100
     print(img, version, scale)