Update blueprints/summarize.py

blueprints/summarize.py

@@ -35,7 +35,7 @@ def get_pipeline(model_name, task_type, hf_token):
     # Load Pipeline
     print(f"Loading model: {model_name}...")
     if task_type == 'SEQ_2_SEQ_LM': # Summarization
-        pipe = pipeline("summarization", model=model_id, device=device)
+        pipe = pipeline("summarization", model=model_name, device=device)
     elif task_type == 'TOKEN_CLS':
         pipe = pipeline("token-classification", model=model_name, aggregation_strategy="simple")
     else: