Update README.md

README.md

@@ -35,18 +35,19 @@ import torch
 import librosa
 from transformers import AutoModel
 
-device = "cuda" 
+device = "cuda" if torch.cuda.is_available() else "cpu"
+torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 
 model = AutoModel.from_pretrained(
     "nectec/Pathumma-llm-audio-1.0.0",
-    torch_dtype=torch.bfloat16, 
+    torch_dtype=torch.bfloat16,
     lora_infer_mode=True,
-    init_from_scratch=True,    
+    init_from_scratch=True,
     trust_remote_code=True
 )
 model = model.to(device)
 
-prompt = "ช่วยถอดความเสียงนี้ให้หน่อย"
+prompt = "ถอดเสียงเป็นข้อความ"
 audio_path = "audio_path.wav"
 audio, sr = librosa.load(audio_path, sr=16000)