Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ The following hyperparameters were used during training:
|
|
80 |
from google.colab import files
|
81 |
|
82 |
# Load the model and processor
|
83 |
-
model_id = "hackergeek98/
|
84 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
85 |
|
86 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(model_id).to(device)
|
|
|
80 |
from google.colab import files
|
81 |
|
82 |
# Load the model and processor
|
83 |
+
model_id = "hackergeek98/whisper-fa-tinyyy"
|
84 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
85 |
|
86 |
model = AutoModelForSpeechSeq2Seq.from_pretrained(model_id).to(device)
|