Add files

.pre-commit-config.yaml

@@ -0,0 +1,46 @@
+exclude: ^stylegan_xl
+repos:
+- repo: https://github.com/pre-commit/pre-commit-hooks
+  rev: v4.2.0
+  hooks:
+  - id: check-executables-have-shebangs
+  - id: check-json
+  - id: check-merge-conflict
+  - id: check-shebang-scripts-are-executable
+  - id: check-toml
+  - id: check-yaml
+  - id: double-quote-string-fixer
+  - id: end-of-file-fixer
+  - id: mixed-line-ending
+    args: ['--fix=lf']
+  - id: requirements-txt-fixer
+  - id: trailing-whitespace
+- repo: https://github.com/myint/docformatter
+  rev: v1.4
+  hooks:
+  - id: docformatter
+    args: ['--in-place']
+- repo: https://github.com/pycqa/isort
+  rev: 5.10.1
+  hooks:
+    - id: isort
+- repo: https://github.com/pre-commit/mirrors-mypy
+  rev: v0.812
+  hooks:
+    - id: mypy
+      args: ['--ignore-missing-imports']
+- repo: https://github.com/google/yapf
+  rev: v0.32.0
+  hooks:
+  - id: yapf
+    args: ['--parallel', '--in-place']
+- repo: https://github.com/kynan/nbstripout
+  rev: 0.5.0
+  hooks:
+    - id: nbstripout
+      args: ['--extra-keys', 'metadata.interpreter metadata.kernelspec cell.metadata.pycharm']
+- repo: https://github.com/nbQA-dev/nbQA
+  rev: 1.3.1
+  hooks:
+    - id: nbqa-isort
+    - id: nbqa-yapf

.style.yapf

@@ -0,0 +1,5 @@
+[style]
+based_on_style = pep8
+blank_line_before_nested_class_or_def = false
+spaces_before_comment = 2
+split_before_logical_operator = true

app.py

@@ -0,0 +1,94 @@
+#!/usr/bin/env python
+
+from __future__ import annotations
+
+import shutil
+import tempfile
+
+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.'
+article = ''
+examples = [
+    [
+        '',
+        'google/vit-base-patch16-224',
+        '',
+        'Demo for google/vit-base-patch16-224',
+        '',
+        '',
+    ],
+    [
+        '',
+        'google/vit-base-patch16-224, microsoft/resnet-50',
+        '',
+        'Compare Image Classification Models',
+        '',
+        '',
+    ],
+    [
+        '',
+        'EleutherAI/gpt-j-6B, EleutherAI/gpt-neo-1.3B',
+        '',
+        'Compare Text Generation Models',
+        '',
+        '',
+    ],
+]
+
+api = HfApi()
+
+
+def save_space_info(dirname: str, filename: str, content: str) -> None:
+    with open(f'{dirname}/{filename}', 'w') as f:
+        f.write(content)
+
+
+def run(space_name: str, model_names_str: str, hf_token: str, title: str,
+        description: str, article: str) -> str:
+    model_names = [name.strip() for name in model_names_str.split(',')]
+    model_names_str = '\n'.join(model_names)
+
+    try:
+        space_url = api.create_repo(repo_id=space_name,
+                                    repo_type='space',
+                                    private=True,
+                                    token=hf_token,
+                                    space_sdk='gradio')
+    except Exception as e:
+        return str(e)
+
+    with tempfile.TemporaryDirectory() as temp_dir:
+        shutil.copy('assets/template.py', f'{temp_dir}/app.py')
+        save_space_info(temp_dir, 'TITLE', title)
+        save_space_info(temp_dir, 'DESCRIPTION', description)
+        save_space_info(temp_dir, 'ARTICLE', article)
+        save_space_info(temp_dir, 'MODEL_NAMES', model_names_str)
+        api.upload_folder(repo_id=space_name,
+                          folder_path=temp_dir,
+                          path_in_repo='.',
+                          token=hf_token,
+                          repo_type='space')
+
+    return f'Successfully created: {space_url}'
+
+
+gr.Interface(
+    fn=run,
+    inputs=[
+        gr.Textbox(label='Space Name'),
+        gr.Textbox(label='Model Names'),
+        gr.Textbox(label='HF TOKEN'),
+        gr.Textbox(label='Title (Optional)'),
+        gr.Textbox(label='Description (Optional)'),
+        gr.Textbox(label='Article (Optional)'),
+    ],
+    outputs=gr.Textbox(label='Output'),
+    title=title,
+    description=description,
+    article=article,
+    examples=examples,
+    cache_examples=False,
+).launch(enable_queue=True, share=False)

assets/template.py

@@ -0,0 +1,36 @@
+#!/usr/bin/env python
+
+from __future__ import annotations
+
+import gradio as gr
+
+
+def read_info(file_name: str) -> str:
+    with open(file_name) as f:
+        content = f.read()
+    return content
+
+
+def load_model(model_name: str) -> gr.Interface:
+    iface = gr.Interface.load(model_name, src='models')
+    for component in iface.output_components:
+        component.label = f'{component.label} ({model_name})'
+    return iface
+
+
+def load_models(model_names: list[str]) -> list[gr.Interface]:
+    return [load_model(name) for name in model_names]
+
+
+title = read_info('TITLE')
+description = read_info('DESCRIPTION')
+article = read_info('ARTICLE')
+model_names = read_info('MODEL_NAMES').split('\n')
+
+interfaces = load_models(model_names)
+gr.Parallel(
+    *interfaces,
+    title=title,
+    description=description,
+    article=article,
+).launch()