Upload README.md
Browse files
README.md
CHANGED
@@ -56,14 +56,14 @@ torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
|
|
56 |
|
57 |
# Load model and processor
|
58 |
model = WhisperForConditionalGeneration.from_pretrained(
|
59 |
-
"
|
60 |
torch_dtype=torch_dtype,
|
61 |
low_cpu_mem_usage=True,
|
62 |
use_safetensors=True
|
63 |
).to(device)
|
64 |
|
65 |
# Load the tokenizer for the model (for decoding)
|
66 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
67 |
|
68 |
# input preprocessing / feature extraction (TBD)
|
69 |
# input_features = ...
|
|
|
56 |
|
57 |
# Load model and processor
|
58 |
model = WhisperForConditionalGeneration.from_pretrained(
|
59 |
+
"mrprimenotes/sign-whisper-german",
|
60 |
torch_dtype=torch_dtype,
|
61 |
low_cpu_mem_usage=True,
|
62 |
use_safetensors=True
|
63 |
).to(device)
|
64 |
|
65 |
# Load the tokenizer for the model (for decoding)
|
66 |
+
tokenizer = AutoTokenizer.from_pretrained("mrprimenotes/sign-whisper-german")
|
67 |
|
68 |
# input preprocessing / feature extraction (TBD)
|
69 |
# input_features = ...
|