my

app.py

@@ -14,8 +14,8 @@ MODELS = {
 }
 
 def load_pipeline(task, model):
-    print(f"Loading pipeline for task: {task} and model: {model}")
-    return pipeline(task, model=model)
+    model_name = MODELS[model]
+    return pipeline(task, model=model_name)
 
 def predict(task, model, text):
     selected_pipeline = load_pipeline(task, model)