Spaces:
Sleeping
Sleeping
Update api.py
Browse files
api.py
CHANGED
|
@@ -64,13 +64,13 @@ def get_model():
|
|
| 64 |
_DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 65 |
|
| 66 |
_MODEL = WhisperForConditionalGeneration.from_pretrained(
|
| 67 |
-
"Kennethdot/
|
| 68 |
torch_dtype=torch.float16 if _DEVICE.type == "cuda" else torch.float32,
|
| 69 |
).to(_DEVICE)
|
| 70 |
_MODEL.eval()
|
| 71 |
|
| 72 |
_PROCESSOR = WhisperProcessor.from_pretrained(
|
| 73 |
-
"Kennethdot/
|
| 74 |
)
|
| 75 |
print(f"Model ready on {_DEVICE}.")
|
| 76 |
return _MODEL, _PROCESSOR, _DEVICE
|
|
|
|
| 64 |
_DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 65 |
|
| 66 |
_MODEL = WhisperForConditionalGeneration.from_pretrained(
|
| 67 |
+
"Kennethdot/kasanoma-whisper-tiny",
|
| 68 |
torch_dtype=torch.float16 if _DEVICE.type == "cuda" else torch.float32,
|
| 69 |
).to(_DEVICE)
|
| 70 |
_MODEL.eval()
|
| 71 |
|
| 72 |
_PROCESSOR = WhisperProcessor.from_pretrained(
|
| 73 |
+
"Kennethdot/kasanoma-whisper-tiny"
|
| 74 |
)
|
| 75 |
print(f"Model ready on {_DEVICE}.")
|
| 76 |
return _MODEL, _PROCESSOR, _DEVICE
|