Spaces:
Running
on
Zero
Running
on
Zero
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=
|
| 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=
|
| 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,
|