Update app.py

app.py

@@ -1,9 +1,11 @@
 
 import os
-os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6"
+os.environ['TORCH_CUDA_ARCH_LIST']="7.5;8.6;9.0;9.0a"
 os.environ["OMP_NUM_THREADS"] = "1"
 os.environ["GRADIO_TEMP_DIR"] = "./gradio_tmp"
 
+import spaces
+
 import os.path as osp
 import torch
 import cv2
@@ -45,6 +47,7 @@ TextModel = MARCONetPlus(
     device=device
 )
 
+@spaces.GPU(duration=120)
 def gradio_inference(input_img, aligned=False, bg_sr=False, scale_factor=2):
     """Run MARCONetPlus inference with optional background SR"""
     if input_img is None: