add handler for wrong model name

.gitignore

@@ -0,0 +1,169 @@
+# (Copy/paste from `transformers`)
+
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# tests and logs
+tests/fixtures/cached_*_text.txt
+logs/
+lightning_logs/
+lang_code_data/
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# vscode
+.vs
+.vscode
+
+# Pycharm
+.idea
+
+# TF code
+tensorflow_code
+
+# Models
+proc_data
+
+# examples
+runs
+/runs_old
+/wandb
+/examples/runs
+/examples/**/*.args
+/examples/rag/sweep
+
+# data
+/data
+serialization_dir
+
+# emacs
+*.*~
+debug.env
+
+# vim
+.*.swp
+
+#ctags
+tags
+
+# pre-commit
+.pre-commit*
+
+# .lock
+*.lock
+
+# DS_Store (MacOS)
+.DS_Store
+
+# ruff
+.ruff_cache

app.py

@@ -72,6 +72,11 @@ DEMO_MARKDOWN = """
 ⛔️ This is still a demo 🤗 Working sections include "Length" and "Quality" ⛔️
 """
 
+MODEL_PROBLEM = """
+😱 Could not retrieve model tags for the specified model, `{model_name}`. Ensure that the model name matches a Hub
+model repo, that it is a public model, and that it has Hub tags.
+"""
+
 SUGGETIONS_HEADER = """
 #### ✨ Here is a list of suggestions for you -- click to expand ✨
 """
@@ -114,7 +119,7 @@ _________________
 # =====================================================================================================================
 def is_valid_task_for_model(model_tags, user_task):
     if len(model_tags) == 0 or user_task == "":
-        return True  # No model / no tags = no problem :)
+        return True  # No model / no task tag = no problem :)
 
     possible_tags = HUB_TAGS[user_task]
     return any(tag in model_tags for tag in possible_tags)
@@ -140,6 +145,10 @@ def get_suggestions(task_type, model_name, problem_type):
     counter = 0
     model_tags = get_model_tags(model_name)
 
+    # If there is a model name but no model tags, something went wrong
+    if model_name != "" and len(model_tags) == 0:
+        return MODEL_PROBLEM.format(model_name=model_name)
+
     user_problem = PROBLEMS.get(problem_type, "")
     user_task = TASK_TYPES.get(task_type, "")