fix bug

automatically choose cpu, gpu settings based on running environment.

controlled_summarization.py

@@ -4,17 +4,21 @@ 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"
+if torch.cuda.is_available():
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+    device = 'gpu'
+    ctrlsum_pipeline = Summarization(os_name="nt",model_name="flan-t5-xl",checkpoint="dyxohjl666/flant5-xl-cocoscisum",device=device)
+else:
+    device = 'cpu'
+    ctrlsum_pipeline = Summarization(os_name="nt",device=device)
 
-ctrlsum_pipeline = Summarization(os_name="nt",model_name="flan-t5-xl",checkpoint="dyxohjl666/flant5-xl-cocoscisum",device=device)
 
 acl_dict = {}
 recommended_kw = {}
+acl_data = load_dataset("dyxohjl666/CocoScisum_ACL", revision="refs/convert/parquet")
 
 
 def convert_to_dict(data):