Check if models are loadable with gradio.Interface.load

app.py

@@ -9,7 +9,9 @@ import gradio as gr
 from huggingface_hub import HfApi
 
 title = 'Model Demo Creation'
-description = 'This is a Space that creates demo apps for models in Model Hub.'
+description = '''
+With this Space, you can create a demo Space for models that are loadable with `gradio.Interface.load` in Model Hub.
+'''
 article = ''
 examples = [
     [
@@ -46,12 +48,12 @@ def check_if_model_exists(model_name: str) -> bool:
                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 check_if_model_loadable(model_name: str) -> bool:
+    try:
+        gr.Interface.load(model_name, src='models')
+    except Exception:
+        return False
+    return True
 
 
 def save_space_info(dirname: str, filename: str, content: str) -> None:
@@ -66,13 +68,25 @@ def run(space_name: str, model_names_str: str, hf_token: str, title: str,
 
     if len(model_names) == 0:
         return 'Model Name cannot be empty.'
-    missing_models = list_missing_models(model_names)
+
+    missing_models = [
+        name for name in model_names if not check_if_model_exists(name)
+    ]
     if len(missing_models) > 0:
-        message = 'The following model were not found: '
+        message = 'The following models were not found: '
         for model_name in missing_models:
             message += f'\n{model_name}'
         return message
 
+    non_loadable_models = [
+        name for name in model_names if not check_if_model_loadable(name)
+    ]
+    if len(non_loadable_models) > 0:
+        message = 'The following models are not loadable with gradio.Interface.load: '
+        for model_name in non_loadable_models:
+            message += f'\n{model_name}'
+        return message
+
     try:
         space_url = api.create_repo(repo_id=space_name,
                                     repo_type='space',