davanstrien HF Staff commited on
Commit
5db5b18
·
1 Parent(s): 525c4ec
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -7,6 +7,7 @@ from typing import List, Optional, Union
7
 
8
  import gradio as gr
9
  import requests
 
10
  from huggingface_hub import (
11
  HfApi,
12
  ModelCard,
@@ -260,7 +261,10 @@ def generate_task_scores_dict():
260
  SCORES = generate_task_scores_dict()
261
 
262
 
263
- @lru_cache(maxsize=None)
 
 
 
264
  def _basic_check(hub_id):
265
  try:
266
  data = ModelMetadata.from_hub(hub_id)
@@ -280,7 +284,7 @@ def _basic_check(hub_id):
280
  score += v["score"]
281
  max_score = task_scores["_max_score"]
282
  score = score / max_score
283
- score_summary = (
284
  f"Your model's metadata score is {round(score*100)}% based on suggested"
285
  f" metadata for {task}. \n"
286
  )
 
7
 
8
  import gradio as gr
9
  import requests
10
+ from diskcache import Cache
11
  from huggingface_hub import (
12
  HfApi,
13
  ModelCard,
 
261
  SCORES = generate_task_scores_dict()
262
 
263
 
264
+ cache = Cache()
265
+
266
+
267
+ @cache.memoize(expire=60 * 60 * 24) # expires after 1 hours
268
  def _basic_check(hub_id):
269
  try:
270
  data = ModelMetadata.from_hub(hub_id)
 
284
  score += v["score"]
285
  max_score = task_scores["_max_score"]
286
  score = score / max_score
287
+ (
288
  f"Your model's metadata score is {round(score*100)}% based on suggested"
289
  f" metadata for {task}. \n"
290
  )