includes ZeroGPU deco for main function

app.py

@@ -1,3 +1,4 @@
+import spaces
 import torch
 import os
 import shutil
@@ -132,6 +133,7 @@ def run_inference(temp_dir, removed_bg_path):
     except subprocess.CalledProcessError as e:
         return f"Error during inference: {str(e)}"
 
+@spaces.GPU(duration=90)
 def process_image(input_pil, remove_bg, progress=gr.Progress(track_tqdm=True)):
 
     torch.cuda.empty_cache()