Check if models exist

app.py

@@ -41,6 +41,19 @@ examples = [
 api = HfApi()
 
 
+def check_if_model_exists(model_name: str) -> bool:
+    return any(info.modelId == model_name
+               for info in api.list_models(search=model_name))
+
+
+def list_missing_models(model_names: list[str]) -> list[str]:
+    missing_models = []
+    for model_name in model_names:
+        if not check_if_model_exists(model_name):
+            missing_models.append(model_name)
+    return missing_models
+
+
 def save_space_info(dirname: str, filename: str, content: str) -> None:
     with open(f'{dirname}/{filename}', 'w') as f:
         f.write(content)
@@ -51,6 +64,15 @@ def run(space_name: str, model_names_str: str, hf_token: str, title: str,
     model_names = [name.strip() for name in model_names_str.split(',')]
     model_names_str = '\n'.join(model_names)
 
+    if len(model_names) == 0:
+        return 'Model Name cannot be empty.'
+    missing_models = list_missing_models(model_names)
+    if len(missing_models) > 0:
+        message = 'The following model were not found: '
+        for model_name in missing_models:
+            message += f'\n{model_name}'
+        return message
+
     try:
         space_url = api.create_repo(repo_id=space_name,
                                     repo_type='space',