Update controlled_summarization.py

controlled_summarization.py

@@ -4,11 +4,14 @@ from SciAssist import Summarization
 import os
 import requests
 from datasets import load_dataset
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+# True
+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",device=device)
+ctrlsum_pipeline = Summarization(os_name="nt",checkpoint="dyxohjl666/flant5-xl-cocoscisum",device=device)
 
 acl_dict = {}
 recommended_kw = {}