Check model inputs/outputs

app.py

@@ -56,6 +56,26 @@ def check_if_model_loadable(model_name: str) -> bool:
     return True
 
 
+def get_model_io_types(
+        model_name: str) -> tuple[tuple[str, ...], tuple[str, ...]]:
+    iface = gr.Interface.load(model_name, src='models')
+    inputs = tuple(map(str, iface.input_components))
+    outputs = tuple(map(str, iface.output_components))
+    return inputs, outputs
+
+
+def check_if_model_io_is_consistent(model_names: list[str]) -> bool:
+    if len(model_names) == 1:
+        return True
+
+    inputs0, outputs0 = get_model_io_types(model_names[0])
+    for name in model_names[1:]:
+        inputs, outputs = get_model_io_types(name)
+        if inputs != inputs0 or outputs != outputs0:
+            return False
+    return True
+
+
 def save_space_info(dirname: str, filename: str, content: str) -> None:
     with open(f'{dirname}/{filename}', 'w') as f:
         f.write(content)
@@ -87,6 +107,9 @@ def run(space_name: str, model_names_str: str, hf_token: str, title: str,
             message += f'\n{model_name}'
         return message
 
+    if not check_if_model_io_is_consistent(model_names):
+        return 'The inputs and outputs of each model must be the same.'
+
     try:
         space_url = api.create_repo(repo_id=space_name,
                                     repo_type='space',