yonikremer commited on
Commit
787d882
1 Parent(s): d2282ae

setting the max cache size to 10

Browse files
Files changed (1) hide show
  1. hanlde_form_submit.py +1 -1
hanlde_form_submit.py CHANGED
@@ -19,7 +19,7 @@ def is_downloaded(model_name: str) -> bool:
19
  return os.path.isdir(model_dir)
20
 
21
 
22
- @lru_cache()
23
  def create_pipeline(model_name: str) -> GroupedSamplingPipeLine:
24
  """
25
  Creates a pipeline with the given model name and group size.
 
19
  return os.path.isdir(model_dir)
20
 
21
 
22
+ @lru_cache(maxsize=10)
23
  def create_pipeline(model_name: str) -> GroupedSamplingPipeLine:
24
  """
25
  Creates a pipeline with the given model name and group size.