Spaces:
Paused
Paused
Update utils.py
Browse files
utils.py
CHANGED
@@ -25,7 +25,7 @@ ARCHITECTURE_MAP = {
|
|
25 |
}
|
26 |
|
27 |
# Function to get the model summary with caching and GPU support
|
28 |
-
@spaces.GPU
|
29 |
@lru_cache(maxsize=10)
|
30 |
def get_model_summary(model_name):
|
31 |
"""
|
|
|
25 |
}
|
26 |
|
27 |
# Function to get the model summary with caching and GPU support
|
28 |
+
@spaces.GPU(duration=120)
|
29 |
@lru_cache(maxsize=10)
|
30 |
def get_model_summary(model_name):
|
31 |
"""
|