Spaces:
Running
on
Zero
Running
on
Zero
Y Phung Nguyen
commited on
Commit
·
09d7494
1
Parent(s):
6698c3b
Fix model preloader
Browse files
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:
|