testing the demo - available models and not all the supported models

tests.py

@@ -1,6 +1,7 @@
 import pytest as pytest
-from grouped_sampling import GroupedSamplingPipeLine, get_full_models_list, UnsupportedModelNameException
+from grouped_sampling import GroupedSamplingPipeLine, UnsupportedModelNameException
 
+from available_models import AVAILABLE_MODELS
 from hanlde_form_submit import create_pipeline, on_form_submit
 
 
@@ -21,7 +22,7 @@ def test_on_form_submit():
 
 @pytest.mark.parametrize(
     "model_name",
-    get_full_models_list()[:3]
+    AVAILABLE_MODELS,
 )
 def test_create_pipeline(model_name: str):
     pipeline: GroupedSamplingPipeLine = create_pipeline(model_name, 5)