Luis J Camargo commited on
Commit
0afb51d
·
1 Parent(s): 3aa360d

import fix

Browse files
Files changed (2) hide show
  1. app.py +1 -0
  2. requirements.txt +1 -0
app.py CHANGED
@@ -5,6 +5,7 @@ import numpy as np
5
  from transformers import WhisperProcessor, AutoConfig, AutoModel, WhisperConfig
6
  from transformers.models.whisper.modeling_whisper import WhisperEncoder
7
  import torch.nn as nn
 
8
  from huggingface_hub import hf_hub_download
9
  import os
10
 
 
5
  from transformers import WhisperProcessor, AutoConfig, AutoModel, WhisperConfig
6
  from transformers.models.whisper.modeling_whisper import WhisperEncoder
7
  import torch.nn as nn
8
+ from safetensors.torch import load_file
9
  from huggingface_hub import hf_hub_download
10
  import os
11
 
requirements.txt CHANGED
@@ -4,3 +4,4 @@ transformers
4
  numpy
5
  librosa
6
  huggingface_hub
 
 
4
  numpy
5
  librosa
6
  huggingface_hub
7
+ safetensors