artificialguybr commited on
Commit
fc61926
β€’
1 Parent(s): 07fdf2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,16 +19,16 @@ import torch
19
  import torchvision
20
  from tqdm import tqdm
21
  from numba import jit
22
- HF_TOKEN = os.environ.get("HF_TOKEN")
23
  from huggingface_hub import HfApi
 
 
 
24
  os.environ["COQUI_TOS_AGREED"] = "1"
25
  api = HfApi(token=HF_TOKEN)
26
  repo_id = "artificialguybr/video-dubbing"
27
-
28
  ZipFile("ffmpeg.zip").extractall()
29
  st = os.stat('ffmpeg')
30
  os.chmod('ffmpeg', st.st_mode | stat.S_IEXEC)
31
-
32
  #Whisper
33
  model_size = "small"
34
  model = WhisperModel(model_size, device="cuda", compute_type="float16")
 
19
  import torchvision
20
  from tqdm import tqdm
21
  from numba import jit
 
22
  from huggingface_hub import HfApi
23
+
24
+
25
+ HF_TOKEN = os.environ.get("HF_TOKEN")
26
  os.environ["COQUI_TOS_AGREED"] = "1"
27
  api = HfApi(token=HF_TOKEN)
28
  repo_id = "artificialguybr/video-dubbing"
 
29
  ZipFile("ffmpeg.zip").extractall()
30
  st = os.stat('ffmpeg')
31
  os.chmod('ffmpeg', st.st_mode | stat.S_IEXEC)
 
32
  #Whisper
33
  model_size = "small"
34
  model = WhisperModel(model_size, device="cuda", compute_type="float16")