Update README.md

README.md

@@ -51,17 +51,17 @@ generation_config.num_return_sequences = 1
 generation_config.pad_token_id = tokenizer.eos_token_id
 generation_config.eos_token_id = tokenizer.eos_token_id
 
-input="xxx"
+input="<Your input, description of the desired building.>"
 input_ids = tokenizer(input, return_tensors="pt", truncation=False).to(device)
 generated_ids = model.generate(input_ids = input_ids.input_ids,
                            attention_mask = input_ids.attention_mask,
                            generation_config = generation_config)
 
-generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
-generated_text = new_tokens.replace("_", " ")
-generated_text = new_tokens.replace("|", "\n")
+generated_output = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
+generated_output = new_tokens.replace("_", " ")
+generated_ooutput = new_tokens.replace("|", "\n")
 
-print(generated_text)
+print(generated_output)
 
 ```