add tag

app.py

@@ -73,6 +73,7 @@ base_model:
     # Add any original YAML fields except base_model
     if original_yaml_header:
         in_base_model_section = False
+        found_tags = False
         for line in original_yaml_header.strip().split('\n'):
             # Skip if we're in a base_model section that continues to the next line
             if in_base_model_section:
@@ -88,8 +89,20 @@ base_model:
                 if ':' in line and len(line.split(':', 1)[1].strip()) > 0:
                     in_base_model_section = False
                 continue
+            
+            # Check for tags field and add bnb-my-repo
+            if line.strip().startswith('tags:'):
+                found_tags = True
+                yaml_header += f"\n{line}"
+                yaml_header += "\n- bnb-my-repo"
+                continue
                 
             yaml_header += f"\n{line}"
+        
+        # If tags field wasn't found, add it
+        if not found_tags:
+            yaml_header += "\ntags:"
+            yaml_header += "\n- bnb-my-repo"
     # Complete the YAML header
     yaml_header += "\n---"
 
@@ -100,7 +113,7 @@ base_model:
 ## Description
 This model is a quantized version of the original model [`{model_name}`](https://huggingface.co/{model_name}). 
 
-It's quantized using the BitsAndBytes library to 4-bit using the [bnb-my-repo](https://huggingface.co/spaces/HF-Quantization/bnb-my-repo) space.
+It's quantized using the BitsAndBytes library to 4-bit using the [bnb-my-repo](https://huggingface.co/spaces/bnb-community/bnb-my-repo) space.
 
 ## Quantization Details
 - **Quantization Type**: int4