Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
ff0fd39
1
Parent(s):
33ae472
fix: zero gpu duration in seconds
Browse files
model.py
CHANGED
|
@@ -17,9 +17,9 @@ def load_model():
|
|
| 17 |
def get_gpu_duration(audio: str) -> int:
|
| 18 |
y, sr = librosa.load(audio)
|
| 19 |
duration = librosa.get_duration(y=y, sr=sr) / 60.0
|
| 20 |
-
gpu_duration = (duration + 59.0) // 60.0
|
| 21 |
print(f"{duration=}, {gpu_duration=}")
|
| 22 |
-
return
|
| 23 |
|
| 24 |
|
| 25 |
@spaces.GPU(duration=get_gpu_duration)
|
|
|
|
| 17 |
def get_gpu_duration(audio: str) -> int:
|
| 18 |
y, sr = librosa.load(audio)
|
| 19 |
duration = librosa.get_duration(y=y, sr=sr) / 60.0
|
| 20 |
+
gpu_duration = max(1.0, (duration + 59.0) // 60.0) * 60.0
|
| 21 |
print(f"{duration=}, {gpu_duration=}")
|
| 22 |
+
return int(gpu_duration)
|
| 23 |
|
| 24 |
|
| 25 |
@spaces.GPU(duration=get_gpu_duration)
|