Spaces:
Runtime error
Runtime error
Commit
·
12c25aa
1
Parent(s):
ea7675e
remove concurrency
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ logging.set_verbosity_error()
|
|
36 |
|
37 |
token = os.getenv("HF_TOKEN")
|
38 |
|
39 |
-
cache = Cache(CACHE_DIR
|
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.
|
561 |
-
|
562 |
-
with gr.
|
563 |
-
|
564 |
-
|
565 |
-
|
566 |
-
|
567 |
-
|
568 |
-
|
569 |
-
|
570 |
-
|
571 |
-
|
572 |
-
|
573 |
-
|
574 |
-
|
575 |
-
|
576 |
-
|
577 |
-
|
578 |
-
|
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:
|