Update README.md

README.md

@@ -28,6 +28,27 @@ Trained on RTX 3070 for 30 hours using SwissDial all Dialects with following gui
 
 ## Uses
 
+
+from peft import PeftModel, PeftConfig
+from transformers import WhisperForConditionalGeneration, Seq2SeqTrainer
+
+peft_model_id = "Flurin17/whisper-large-v3-peft-swiss-german" # Use the same model ID as before.
+peft_config = PeftConfig.from_pretrained(peft_model_id)
+model = WhisperForConditionalGeneration.from_pretrained(
+    peft_config.base_model_name_or_path, load_in_8bit=True, device_map="auto"
+)
+model = PeftModel.from_pretrained(model, peft_model_id)
+model.config.use_cache = True
+
+
+from transformers import WhisperFeatureExtractor
+
+feature_extractor = WhisperFeatureExtractor.from_pretrained(peft_model_id)
+
+from transformers import WhisperTokenizer
+
+tokenizer = WhisperTokenizer.from_pretrained(peft_model_id, language=language, task=task)
+
 from transformers import AutomaticSpeechRecognitionPipeline
 import torch
 pipe = AutomaticSpeechRecognitionPipeline(model=model, tokenizer=tokenizer, feature_extractor=feature_extractor)