Update app.py

app.py

@@ -4,6 +4,7 @@ import functools
 import os
 import tempfile
 import torch
+import spaces
 import gradio as gr
 from PIL import Image
 from gradio_imageslider import ImageSlider
@@ -52,9 +53,9 @@ def create_demo():
     predictor = load_predictor()
     
     # Create processing functions for each data type
-    process_object = functools.partial(process_image, predictor, data_type="object")
-    process_scene = functools.partial(process_image, predictor, data_type="indoor")
-    process_human = functools.partial(process_image, predictor, data_type="object")
+    process_object = spaces.GPU(functools.partial(process_image, predictor, data_type="object"))
+    process_scene = spaces.GPU(functools.partial(process_image, predictor, data_type="indoor"))
+    process_human = spaces.GPU(functools.partial(process_image, predictor, data_type="object"))
 
     # Define markdown content
     HEADER_MD = """