Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -18,14 +18,7 @@ from transformers.pipelines.audio_utils import ffmpeg_read
|
|
18 |
import torch
|
19 |
from datasets import load_dataset, Dataset, DatasetDict
|
20 |
import spaces
|
21 |
-
import subprocess
|
22 |
|
23 |
-
# Install Flash Attention if possible
|
24 |
-
subprocess.run(
|
25 |
-
"pip install flash-attn --no-build-isolation",
|
26 |
-
env={"FLASH_ATTENTION_SKIP_CUDA_BUILD": "TRUE"},
|
27 |
-
shell=True,
|
28 |
-
)
|
29 |
|
30 |
# Constants
|
31 |
MODEL_NAME = "openai/whisper-large-v3-turbo"
|
|
|
18 |
import torch
|
19 |
from datasets import load_dataset, Dataset, DatasetDict
|
20 |
import spaces
|
|
|
21 |
|
|
|
|
|
|
|
|
|
|
|
|
|
22 |
|
23 |
# Constants
|
24 |
MODEL_NAME = "openai/whisper-large-v3-turbo"
|