Spaces:
Sleeping
Sleeping
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)
|