Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -245,7 +245,7 @@ class ImageDataset(Dataset):
|
|
| 245 |
return self.transform(img), os.path.basename(img_path)
|
| 246 |
|
| 247 |
def measure_duration(images, threshold) -> int:
|
| 248 |
-
return ceil(len(images) / 64) *
|
| 249 |
|
| 250 |
@dynGPU(duration=measure_duration)
|
| 251 |
def process_images(images, threshold):
|
|
|
|
| 245 |
return self.transform(img), os.path.basename(img_path)
|
| 246 |
|
| 247 |
def measure_duration(images, threshold) -> int:
|
| 248 |
+
return ceil(len(images) / 64) * 5 + 3
|
| 249 |
|
| 250 |
@dynGPU(duration=measure_duration)
|
| 251 |
def process_images(images, threshold):
|