Spaces:
Runtime error
Runtime error
Prevent preallocation of gradio interface, do it at runtime
Browse files
app.py
CHANGED
|
@@ -95,8 +95,8 @@ SUPPORTED_MODEL_NAMES = sorted(list(SUPPORTED_MODEL_NAMES))
|
|
| 95 |
model_dict = {}
|
| 96 |
for model_name in SUPPORTED_MODEL_NAMES:
|
| 97 |
try:
|
| 98 |
-
iface = gr.Interface.load(f'models/{model_name}')
|
| 99 |
-
model_dict[model_name] =
|
| 100 |
except:
|
| 101 |
pass
|
| 102 |
|
|
@@ -370,6 +370,17 @@ def infer_audio(model_name: str, audio_file: str) -> str:
|
|
| 370 |
# Obtain Gradio Model function from cache of models
|
| 371 |
if model_name in model_dict:
|
| 372 |
model = model_dict[model_name]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 373 |
else:
|
| 374 |
model = None
|
| 375 |
|
|
|
|
| 95 |
model_dict = {}
|
| 96 |
for model_name in SUPPORTED_MODEL_NAMES:
|
| 97 |
try:
|
| 98 |
+
# iface = gr.Interface.load(f'models/{model_name}')
|
| 99 |
+
model_dict[model_name] = None
|
| 100 |
except:
|
| 101 |
pass
|
| 102 |
|
|
|
|
| 370 |
# Obtain Gradio Model function from cache of models
|
| 371 |
if model_name in model_dict:
|
| 372 |
model = model_dict[model_name]
|
| 373 |
+
|
| 374 |
+
if model is None:
|
| 375 |
+
# Load the gradio interface
|
| 376 |
+
try:
|
| 377 |
+
iface = gr.Interface.load(f'models/{model_name}')
|
| 378 |
+
except:
|
| 379 |
+
iface = None
|
| 380 |
+
|
| 381 |
+
if iface is not None:
|
| 382 |
+
# Update model cache
|
| 383 |
+
model_dict[model_name] = iface
|
| 384 |
else:
|
| 385 |
model = None
|
| 386 |
|