JacobLinCool commited on
Commit
ff0fd39
·
1 Parent(s): 33ae472

fix: zero gpu duration in seconds

Browse files
Files changed (1) hide show
  1. model.py +2 -2
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 max(1, int(gpu_duration))
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)