dwb2023 commited on
Commit
dbfd6b3
1 Parent(s): f70face

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
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
  """