show loading model only when loading

app.py

@@ -16,11 +16,12 @@ default_model_name = "gia-project/gia2-small-untrained"
 
 
 def generate_text(model_name, input_text):
-    # Inform the user that the model is loading
-    yield "Loading model..."
+
 
     # Check if the selected model is already loaded
     if model_name not in model_pipelines:
+        # Inform the user that the model is loading
+        yield "Loading model..."
         # Load the model and create a pipeline if it's not already loaded
         generator = pipeline("text-generation", model=model_name, trust_remote_code=True)
         model_pipelines[model_name] = generator