derektan commited on
Commit
ee3a9da
Β·
1 Parent(s): 97c5862

Changed duration all to 20s

Browse files
Taxabind/TaxaBind/SatBind/clip_seg_tta.py CHANGED
@@ -169,7 +169,7 @@ class ClipSegTTA:
169
  )
170
 
171
  # integration with ZeroGPU on hf
172
- @spaces.GPU(duration=5)
173
  def reset(self, sample_idx):
174
  """Reset the parameters & local model for the current sample."""
175
  if self.load_model:
@@ -260,7 +260,7 @@ class ClipSegTTA:
260
  # print(f"Local model reset in {(time.time()-start_time):.2f}s.")
261
 
262
  # # integration with ZeroGPU on hf
263
- @spaces.GPU(duration=5)
264
  def execute_tta(
265
  self,
266
  patch_indices: list,
 
169
  )
170
 
171
  # integration with ZeroGPU on hf
172
+ @spaces.GPU(duration=20)
173
  def reset(self, sample_idx):
174
  """Reset the parameters & local model for the current sample."""
175
  if self.load_model:
 
260
  # print(f"Local model reset in {(time.time()-start_time):.2f}s.")
261
 
262
  # # integration with ZeroGPU on hf
263
+ @spaces.GPU(duration=20)
264
  def execute_tta(
265
  self,
266
  patch_indices: list,
app_multimodal_inference.py CHANGED
@@ -149,7 +149,7 @@ def _array_to_pil(arr: np.ndarray) -> Image.Image:
149
 
150
  # ────────────────────────── main inference ────────────────────────────
151
  # integration with ZeroGPU on hf
152
- @spaces.GPU
153
  def process(
154
  sat_img: Image.Image,
155
  taxonomy: str,
 
149
 
150
  # ────────────────────────── main inference ────────────────────────────
151
  # integration with ZeroGPU on hf
152
+ @spaces.GPU(duration=20)
153
  def process(
154
  sat_img: Image.Image,
155
  taxonomy: str,