Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Commit
Β·
49904fc
1
Parent(s):
a0e66b8
Update dataset repository ID and add model presence validation in submission process
Browse files
utils.py
CHANGED
@@ -1,15 +1,15 @@
|
|
1 |
import gradio as gr
|
2 |
import pandas as pd
|
3 |
import json
|
4 |
-
import os
|
5 |
from pathlib import Path
|
6 |
from huggingface_hub import HfApi
|
|
|
7 |
from datasets import load_dataset
|
8 |
|
9 |
api = HfApi()
|
10 |
|
11 |
OWNER = "Navid-AI"
|
12 |
-
DATASET_REPO_ID = f"{OWNER}/requests-dataset"
|
13 |
|
14 |
results_dir = Path(__file__).parent / "results"
|
15 |
|
@@ -87,6 +87,15 @@ def submit_model(model_name, revision, precision, params, license, task, model_p
|
|
87 |
df_finished = load_requests("finished", task_type=task)
|
88 |
df_failed = load_requests("failed", task_type=task)
|
89 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
90 |
# Check if Auto Fetch feature couldn't fetch model info
|
91 |
if params == "N/A":
|
92 |
return (
|
|
|
1 |
import gradio as gr
|
2 |
import pandas as pd
|
3 |
import json
|
|
|
4 |
from pathlib import Path
|
5 |
from huggingface_hub import HfApi
|
6 |
+
from huggingface_hub.errors import RepositoryNotFoundError
|
7 |
from datasets import load_dataset
|
8 |
|
9 |
api = HfApi()
|
10 |
|
11 |
OWNER = "Navid-AI"
|
12 |
+
DATASET_REPO_ID = f"{OWNER}/requests-dataset-rag"
|
13 |
|
14 |
results_dir = Path(__file__).parent / "results"
|
15 |
|
|
|
87 |
df_finished = load_requests("finished", task_type=task)
|
88 |
df_failed = load_requests("failed", task_type=task)
|
89 |
|
90 |
+
# Validate model presence
|
91 |
+
try:
|
92 |
+
api.model_info(model_name)
|
93 |
+
except RepositoryNotFoundError:
|
94 |
+
return (
|
95 |
+
f"<h2 style='color:red; text-align:center;'>β Model '{model_name}' not found on HuggingFace Hub.</h2>",
|
96 |
+
df_pending,
|
97 |
+
)
|
98 |
+
|
99 |
# Check if Auto Fetch feature couldn't fetch model info
|
100 |
if params == "N/A":
|
101 |
return (
|