Add pipeline tag and library name

This PR improves the model card by adding a relevant pipeline tag and library name.

README.md

@@ -1,7 +1,9 @@
 ---
-license: apache-2.0
 base_model:
 - google/gemma-2-9b-it
+license: apache-2.0
+pipeline_tag: text-generation
+library_name: transformers
 ---
 
 # General Preference Representation Model (GPM)
@@ -131,8 +133,7 @@ def get_reward_model(base_causal_model, base_llm_model, value_head_dim: int, add
                 block_values = self.prompt_head(prompt_hidden_states).view(batch_size, dim // 2)
                 block_values = torch.softmax(block_values, dim=-1)
                 
-                # Create a batch of zero matrices [batch_size, dim, dim]
-                batch_R_matrices = torch.zeros((batch_size, dim, dim), device=device, dtype=dtype)
+                # Create a batch of zero matrices [batch_size, dim, dim]\n                batch_R_matrices = torch.zeros((batch_size, dim, dim), device=device, dtype=dtype)
                 
                 # Fill only the block diagonal entries with the learned values
                 for i in range(0, dim, 2):