Kennethdot commited on
Commit
cf13755
·
verified ·
1 Parent(s): 033bc68

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +2 -2
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/kasanoma_whisper",
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"
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