Spaces:
Runtime error
Runtime error
BjarneBepaData
commited on
Commit
•
8eb8554
1
Parent(s):
cf78cd2
Another try
Browse files- 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",
|