added a supported model check

hanlde_form_submit.py

@@ -6,6 +6,8 @@ from grouped_sampling import GroupedSamplingPipeLine
 
 from download_repo import download_repository
 from prompt_engeneering import rewrite_prompt
+from supported_models import is_supported, SUPPORTED_MODEL_NAME_PAGES_FORMAT, BLACKLISTED_MODEL_NAMES, \
+    BLACKLISTED_ORGANIZATIONS
 
 
 def is_downloaded(model_name: str) -> bool:
@@ -93,6 +95,15 @@ def on_form_submit(
      TypeError: If the output length is not an integer or the prompt is not a string.
      RuntimeError: If the model is not found.
     """
+    if not is_supported(model_name, 1, 1):
+        raise ValueError(
+            f"The model: {model_name} is not supported."
+            f"The supported models are the models from {SUPPORTED_MODEL_NAME_PAGES_FORMAT}"
+            f" that satisfy the following conditions:\n"
+            f"1. The model has at least one like and one download.\n"
+            f"2. The model is not one of: {BLACKLISTED_MODEL_NAMES}.\n"
+            f"3. The model was not created any of those organizations: {BLACKLISTED_ORGANIZATIONS}.\n"
+        )
     if len(prompt) == 0:
         raise ValueError(f"The prompt must not be empty.")
     st.write(f"Loading model: {model_name}...")

supported_models.py

@@ -1,3 +1,4 @@
+from functools import lru_cache
 from typing import Generator, Set, Union, List, Optional
 
 import requests
@@ -137,14 +138,28 @@ def generate_supported_model_names(
                 )
 
 
+@lru_cache
 def get_supported_model_names(
         min_number_of_downloads: int = DEFAULT_MIN_NUMBER_OF_DOWNLOADS,
         min_number_of_likes: int = DEFAULT_MIN_NUMBER_OF_LIKES,
 ) -> Set[str]:
-    return set(generate_supported_model_names(
+    return set(
+        generate_supported_model_names(
+            min_number_of_downloads=min_number_of_downloads,
+            min_number_of_likes=min_number_of_likes,
+        )
+    )
+
+
+def is_supported(
+        model_name: str,
+        min_number_of_downloads: int = DEFAULT_MIN_NUMBER_OF_DOWNLOADS,
+        min_number_of_likes: int = DEFAULT_MIN_NUMBER_OF_LIKES,
+) -> bool:
+    return model_name in get_supported_model_names(
         min_number_of_downloads=min_number_of_downloads,
         min_number_of_likes=min_number_of_likes,
-    ))
+    )
 
 
 if __name__ == "__main__":