Update app.py

app.py

@@ -3,6 +3,8 @@ import sys
 from torchvision.transforms import functional
 sys.modules["torchvision.transforms.functional_tensor"] = functional
 
+import spaces
+
 from basicsr.archs.srvgg_arch import SRVGGNetCompact
 from gfpgan.utils import GFPGANer
 from realesrgan.utils import RealESRGANer
@@ -33,6 +35,7 @@ upsampler = RealESRGANer(scale=4, model_path=model_path, model=model, tile=0, ti
 # 이미지 저장 디렉토리 생성 (필요시 주석 해제)
 # os.makedirs('output', exist_ok=True)
 
+@spaces.GPU(duration=120)
 def upscaler(img, version, scale):
     try:
         # 입력된 img는 파일 경로임