Update controlled_summarization.py

controlled_summarization.py

@@ -11,7 +11,7 @@ print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
 acl_data = load_dataset("dyxohjl666/CocoScisum_ACL", revision="refs/convert/parquet")
 device = "gpu" if torch.cuda.is_available() else "cpu"
 
-ctrlsum_pipeline = Summarization(os_name="nt",checkpoint="dyxohjl666/flant5-xl-cocoscisum",device=device)
+ctrlsum_pipeline = Summarization(os_name="nt",model_name="flan-t5-xl",checkpoint="dyxohjl666/flant5-xl-cocoscisum",device=device)
 
 acl_dict = {}
 recommended_kw = {}