DavidLoccus commited on
Commit
2f2dad0
·
1 Parent(s): 6e005ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -1
app.py CHANGED
@@ -7,10 +7,18 @@ import numpy as np
7
  from pytube import YouTube
8
  import os
9
  import random
 
10
 
11
  FS=16000
12
  MAX_SIZE = FS * 30
13
- MODEL = torch.jit.load(MODELNAME)
 
 
 
 
 
 
 
14
 
15
 
16
  def reformat_freq(sr, y):
 
7
  from pytube import YouTube
8
  import os
9
  import random
10
+ from huggingface_hub import HfApi
11
 
12
  FS=16000
13
  MAX_SIZE = FS * 30
14
+
15
+ hf_api = HfApi(
16
+ endpoint="https://huggingface.co", # Can be a Private Hub endpoint.
17
+ token=HF_TOKEN_DEMO, # Token is not persisted on the machine.
18
+ )
19
+
20
+ modelfile = hf_api.hf_hub_download(MODEL_REPO,MODELNAME)
21
+ MODEL = torch.jit.load(modelfile)
22
 
23
 
24
  def reformat_freq(sr, y):