MohamedRashad commited on
Commit
49904fc
Β·
1 Parent(s): a0e66b8

Update dataset repository ID and add model presence validation in submission process

Browse files
Files changed (1) hide show
  1. utils.py +11 -2
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 (