Spaces:
Running
on
Zero
Running
on
Zero
add more duration
Browse files
app.py
CHANGED
|
@@ -50,8 +50,7 @@ class calculateDuration:
|
|
| 50 |
print(f"Activity: {self.activity_name}, End time: {self.start_time_formatted}")
|
| 51 |
|
| 52 |
|
| 53 |
-
|
| 54 |
-
@spaces.GPU(duration=20)
|
| 55 |
@torch.inference_mode()
|
| 56 |
@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
| 57 |
def process_image(image_input, image_url, task_prompt, text_prompt=None, dilate=0, merge_masks=False, return_rectangles=False, progress=gr.Progress(track_tqdm=True)) -> Optional[Image.Image]:
|
|
|
|
| 50 |
print(f"Activity: {self.activity_name}, End time: {self.start_time_formatted}")
|
| 51 |
|
| 52 |
|
| 53 |
+
@spaces.GPU()
|
|
|
|
| 54 |
@torch.inference_mode()
|
| 55 |
@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
|
| 56 |
def process_image(image_input, image_url, task_prompt, text_prompt=None, dilate=0, merge_masks=False, return_rectangles=False, progress=gr.Progress(track_tqdm=True)) -> Optional[Image.Image]:
|