Integrated with hf Zero GPU

app.py

@@ -21,6 +21,7 @@ import os, glob, threading, time
 import torch
 from PIL import Image
 import json
+import spaces   # integration with ZeroGPU on hf
 
 # Import configuration & RL / TTA utilities -------------------------------------------------
 # NOTE: we import * so that the global names (e.g. USE_GPU, MODEL_NAME, etc.)
@@ -137,7 +138,8 @@ def process_no_tta(
 
 
 
-
+# integration with ZeroGPU on hf
+@spaces.GPU
 def process(
     sat_path: str | None,
     ground_path: str | None,

test_parameter.py

@@ -162,7 +162,7 @@ INPUT_DIM = 4
 EMBEDDING_DIM = 128
 K_SIZE = 8   # 8
 
-USE_GPU = False  # do you want to use GPUS?
+USE_GPU = True  # do you want to use GPUS?
 NUM_GPU = getenv("NUM_GPU", default=2, cast_type=int) # the number of GPUs
 NUM_META_AGENT = getenv("NUM_META_AGENT", default=4, cast_type=int)  # the number of processes
 FOLDER_NAME = 'inference'