davanstrien HF Staff commited on
Commit
12c25aa
·
1 Parent(s): ea7675e

remove concurrency

Browse files
Files changed (1) hide show
  1. app.py +33 -24
app.py CHANGED
@@ -36,7 +36,7 @@ logging.set_verbosity_error()
36
 
37
  token = os.getenv("HF_TOKEN")
38
 
39
- cache = Cache(CACHE_DIR, eviction_policy="least-frequently-used", size_limit=4e9)
40
 
41
  def get_model_labels(model):
42
  try:
@@ -290,7 +290,7 @@ SCORES = generate_task_scores_dict()
290
  GENERIC_SCORES = generate_common_scores()
291
 
292
 
293
- @cache.memoize(expire=60 * 60 * 24 * 3) # expires after 3 days
294
  def _basic_check(hub_id):
295
  data = ModelMetadata.from_hub(hub_id)
296
  score = 0
@@ -351,7 +351,7 @@ def create_query_url(query, skip=0):
351
  return f"https://huggingface.co/api/search/full-text?q={query}&limit=100&skip={skip}&type=model"
352
 
353
 
354
- @cache.memoize(expire=60 * 60 * 24 * 3) # expires after 3 days
355
  def get_results(query) -> Dict[Any, Any]:
356
  url = create_query_url(query)
357
  r = httpx.get(url)
@@ -557,29 +557,38 @@ def search_hub(query: str, min_score=None, min_model_card_length=None):
557
 
558
 
559
  with gr.Blocks() as demo:
560
- gr.Markdown("# 🤗 Hub model search with metadata quality filters")
561
- with gr.Row():
562
- with gr.Column():
563
- query = gr.Textbox("x-ray", label="Search query")
564
- with gr.Column():
565
- button = gr.Button("Search")
566
- with gr.Row():
567
- # gr.Checkbox(False, label="Must have licence?")
568
- mim_model_card_length = gr.Number(
569
- None, label="Minimum model card length"
570
- )
571
- min_metadata_score = gr.Slider(0, label="Minimum metadata score")
572
- filter_results = gr.Markdown("Filter results vs original search")
573
- results_markdown = gr.Markdown("Search results")
574
-
575
- button.click(
576
- search_hub,
577
- [query, min_metadata_score, mim_model_card_length],
578
- [filter_results, results_markdown],
579
- )
 
 
 
 
 
 
 
 
 
580
 
581
 
582
- demo.launch()
583
 
584
 
585
  # with gr.Blocks() as demo:
 
36
 
37
  token = os.getenv("HF_TOKEN")
38
 
39
+ cache = Cache(CACHE_DIR)
40
 
41
  def get_model_labels(model):
42
  try:
 
290
  GENERIC_SCORES = generate_common_scores()
291
 
292
 
293
+ # @cache.memoize(expire=60 * 60 * 24 * 3) # expires after 3 days
294
  def _basic_check(hub_id):
295
  data = ModelMetadata.from_hub(hub_id)
296
  score = 0
 
351
  return f"https://huggingface.co/api/search/full-text?q={query}&limit=100&skip={skip}&type=model"
352
 
353
 
354
+ # @cache.memoize(expire=60 * 60 * 24 * 3) # expires after 3 days
355
  def get_results(query) -> Dict[Any, Any]:
356
  url = create_query_url(query)
357
  r = httpx.get(url)
 
557
 
558
 
559
  with gr.Blocks() as demo:
560
+ with gr.Tab("Hub search with metadata quality filter"):
561
+ gr.Markdown("# 🤗 Hub model search with metadata quality filters")
562
+ with gr.Row():
563
+ with gr.Column():
564
+ query = gr.Textbox("x-ray", label="Search query")
565
+ with gr.Column():
566
+ button = gr.Button("Search")
567
+ with gr.Row():
568
+ # gr.Checkbox(False, label="Must have licence?")
569
+ mim_model_card_length = gr.Number(
570
+ None, label="Minimum model card length"
571
+ )
572
+ min_metadata_score = gr.Slider(0, label="Minimum metadata score")
573
+ filter_results = gr.Markdown("Filter results vs original search")
574
+ results_markdown = gr.Markdown("Search results")
575
+
576
+ button.click(
577
+ search_hub,
578
+ [query, min_metadata_score, mim_model_card_length],
579
+ [filter_results, results_markdown],
580
+ )
581
+ with gr.Tab("Scoring metadata quality"):
582
+ with gr.Row():
583
+ gr.Markdown(f"""
584
+ # Metadata quality scoring
585
+ ```
586
+ {COMMON_SCORES}
587
+ ```
588
+ """)
589
 
590
 
591
+ demo.launch(prevent_thread_lock=True)
592
 
593
 
594
  # with gr.Blocks() as demo: