Update README.md

README.md

@@ -5,6 +5,11 @@ tags:
 - quantization
 - 4bit
 - lora
+license: apache-2.0
+base_model: google/gemma-2-2b
+datasets:
+- medicaltranscriptions
+pipeline_tag: text-generation
 ---
 
 # Model Card for Medical Transcription Model (Gemma-MedTr)
@@ -45,4 +50,3 @@ bnb_config = BitsAndBytesConfig(
 
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=access_token_read)
 model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config, device_map='auto', token=access_token_read)
-'''