BjarneBepaData commited on
Commit
8eb8554
1 Parent(s): cf78cd2

Another try

Browse files
Files changed (1) hide show
  1. app/main.py +2 -2
app/main.py CHANGED
@@ -10,11 +10,11 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
10
  model_id = "openai/whisper-large-v3"
11
 
12
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
13
- model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
14
  )
15
  model.to(device)
16
 
17
- processor = AutoProcessor.from_pretrained(model_id)
18
 
19
  pipe = pipeline(
20
  "automatic-speech-recognition",
 
10
  model_id = "openai/whisper-large-v3"
11
 
12
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
13
+ model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True, cache_dir="./app.cache"
14
  )
15
  model.to(device)
16
 
17
+ processor = AutoProcessor.from_pretrained(model_id, cache_dir="./app.cache")
18
 
19
  pipe = pipeline(
20
  "automatic-speech-recognition",