Added artist names to UI

model_config.yaml

@@ -3,7 +3,7 @@ Rap:
   artist_names: 
   - Eminem
   - Hopsin
-  - Kentrick Lamar
+  - Kendrick Lamar
   mlflow_run_id: 16c4ff05d92a45d79d89572a58b6424b
   hf_commit_hash: ca066f322213fbeac8d036fafd32112e23837722
   task: text-generation

src/utils.py

@@ -8,7 +8,8 @@ def load_pipelines_from_config(config_path: str):
 
     models = {}
     for model, config in model_config.items():
-        models[model] = pipeline(
+        name = f"{model} - ({', '.join(config['artist_names'])})"
+        models[name] = pipeline(
             task=config["task"],
             model=config["model_name"],
             revision=config["hf_commit_hash"],