Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import subprocess
|
|
2 |
|
3 |
subprocess.run(["pip", "install", "datasets"])
|
4 |
subprocess.run(["pip", "install", "transformers"])
|
5 |
-
subprocess.run(["pip", "install", "torch
|
6 |
|
7 |
from transformers import WhisperProcessor, WhisperForConditionalGeneration
|
8 |
from datasets import load_dataset
|
|
|
2 |
|
3 |
subprocess.run(["pip", "install", "datasets"])
|
4 |
subprocess.run(["pip", "install", "transformers"])
|
5 |
+
subprocess.run(["pip", "install", "torch", "torchvision", "torchaudio", "-f", "https://download.pytorch.org/whl/torch_stable.html"])
|
6 |
|
7 |
from transformers import WhisperProcessor, WhisperForConditionalGeneration
|
8 |
from datasets import load_dataset
|