Update README.md

README.md

@@ -1,17 +1,21 @@
-library_name: transformers
+---
+license: apache-2.0
+datasets:
+- harishnair04/mtsamples
+language:
+- en
+base_model:
+- google/gemma-2-2b
+pipeline_tag: text-generation
 tags:
 - trl
 - sft
 - quantization
 - 4bit
-- lora
-license: apache-2.0
-base_model: google/gemma-2-2b
-datasets:
-- medicaltranscriptions
-pipeline_tag: text-generation
+- LoRA
 ---
 
+
 # Model Card for Medical Transcription Model (Gemma-MedTr)
 
 This model is a fine-tuned variant of `Gemma-2-2b`, optimized for medical transcription tasks with efficient 4-bit quantization and Low-Rank Adaptation (LoRA). It handles transcription processing, keyword extraction, and medical specialty classification.
@@ -49,4 +53,4 @@ 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)
+model = AutoModelForCausalLM.from_pretrained(model_id, quantization_config=bnb_config, device_map='auto', token=access_token_read)