dwb2023 commited on
Commit
8bf0b67
·
verified ·
1 Parent(s): ad6330a

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -0
utils.py CHANGED
@@ -19,6 +19,7 @@ def authenticate_hf(token):
19
  # Function to get the model summary
20
  model_cache = {}
21
 
 
22
  def get_model_summary(model_name):
23
  if model_name in model_cache:
24
  return model_cache[model_name], ""
 
19
  # Function to get the model summary
20
  model_cache = {}
21
 
22
+ @spaces.GPU
23
  def get_model_summary(model_name):
24
  if model_name in model_cache:
25
  return model_cache[model_name], ""