testing using small models

tests.py

@@ -1,7 +1,6 @@
 import pytest as pytest
 from grouped_sampling import GroupedSamplingPipeLine, UnsupportedModelNameException
 
-from available_models import AVAILABLE_MODELS
 from hanlde_form_submit import create_pipeline, on_form_submit
 
 
@@ -20,11 +19,8 @@ def test_on_form_submit():
         on_form_submit(unsupported_model_name, output_length, prompt)
 
 
-@pytest.mark.parametrize(
-    "model_name",
-    AVAILABLE_MODELS,
-)
-def test_create_pipeline(model_name: str):
+def test_create_pipeline():
+    model_name = "gpt2"
     pipeline: GroupedSamplingPipeLine = create_pipeline(model_name, 5)
     assert pipeline is not None
     assert pipeline.model_name == model_name