Update README.md

README.md

@@ -78,7 +78,7 @@ def query_model(input_text):
     """Query the model and get a generated response."""
     formatted_input = f"<|title|>{input_text}<|prompt|>"
     input_ids = tokenizer.encode(formatted_input, return_tensors="pt")
-    attention_mask = torch.ones(input_ids.shape, dtype=torch.long)  # create attention mask of 1s
+    attention_mask = torch.ones(input_ids.shape, dtype=torch.long)
 
     # Generate a response
     output = model.generate(input_ids, attention_mask=attention_mask, max_length=256,