Emmanuel08 commited on
Commit
e55e420
·
verified ·
1 Parent(s): cca59f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ device = "cpu"
10
  torch_dtype = torch.float32 # Use CPU-friendly float type
11
  MODEL_NAME = "openai/whisper-tiny" # ✅ Switched to smallest model for fastest performance
12
 
13
- # ✅ 2️⃣ Load Whisper Tiny Model on CPU
14
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
15
- MODEL_NAME, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
16
  )
17
  model.to(device)
18
 
 
10
  torch_dtype = torch.float32 # Use CPU-friendly float type
11
  MODEL_NAME = "openai/whisper-tiny" # ✅ Switched to smallest model for fastest performance
12
 
13
+ # ✅ 2️⃣ Load Whisper Tiny Model on CPU (Removed `low_cpu_mem_usage=True`)
14
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
15
+ MODEL_NAME, torch_dtype=torch_dtype, use_safetensors=True # ✅ Removed low_cpu_mem_usage
16
  )
17
  model.to(device)
18