Y Phung Nguyen commited on
Commit
09d7494
·
1 Parent(s): 6698c3b

Fix model preloader

Browse files
Files changed (1) hide show
  1. ui.py +2 -2
ui.py CHANGED
@@ -293,7 +293,7 @@ def create_demo():
293
  )
294
 
295
  # GPU-decorated function to load any model (for user selection)
296
- @spaces.GPU(max_duration=MAX_DURATION)
297
  def load_model_with_gpu(model_name):
298
  """Load medical model (GPU-decorated for ZeroGPU compatibility)"""
299
  try:
@@ -549,7 +549,7 @@ def create_demo():
549
  )
550
 
551
  # GPU-decorated function to load Whisper ASR model on-demand
552
- @spaces.GPU(max_duration=MAX_DURATION)
553
  def load_whisper_model_on_demand():
554
  """Load Whisper ASR model when needed"""
555
  try:
 
293
  )
294
 
295
  # GPU-decorated function to load any model (for user selection)
296
+ # @spaces.GPU(max_duration=MAX_DURATION)
297
  def load_model_with_gpu(model_name):
298
  """Load medical model (GPU-decorated for ZeroGPU compatibility)"""
299
  try:
 
549
  )
550
 
551
  # GPU-decorated function to load Whisper ASR model on-demand
552
+ # @spaces.GPU(max_duration=MAX_DURATION)
553
  def load_whisper_model_on_demand():
554
  """Load Whisper ASR model when needed"""
555
  try: