Update README.md

README.md

@@ -43,11 +43,11 @@ from transformers import (
     AutoTokenizer,
 )
 
-# Input the rest port of IDF file.
+# Load the rest port of IDF file.
 file_path = "v1_nextpart.idf"
 output_path = "v1_final.idf"
 
-# Input the EPlus-LLM model
+# Load the EPlus-LLM model
 tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
 model = AutoModelForSeq2SeqLM.from_pretrained("EPlus-LLM/EPlus-LLMv1")
 
@@ -61,7 +61,7 @@ generation_config.pad_token_id = tokenizer.eos_token_id
 generation_config.eos_token_id = tokenizer.eos_token_id
 
 # Please provide your input here — a description of the desired building
-# For more details, please refer to the paper
+# For more details, please refer to the paper: https://doi.org/10.1016/j.apenergy.2024.123431
 input="Simulate a building that is 30.00 meters long, 15.00 meters wide, and 3.50 meters high. The window-to-wall ratio is 0.28. The occupancy rate is 8.00 m2/people, the lighting level is 6.00 W/m2, and the equipment power consumption is 8.80 W/m2."
 input_ids = tokenizer(input, return_tensors="pt", truncation=False)
 generated_ids = model.generate(input_ids = input_ids.input_ids,