Kushtrim commited on
Commit
b58c31f
·
verified ·
1 Parent(s): b646ce4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from transformers import pipeline, AutoModelForSpeechSeq2Seq, AutoProcessor
2
  from transformers.pipelines.audio_utils import ffmpeg_read
 
3
  import yt_dlp as youtube_dl
4
  import gradio as gr
5
  import tempfile
@@ -7,6 +8,8 @@ import torch
7
  import time
8
  import os
9
 
 
 
10
  BATCH_SIZE = 16
11
  FILE_LIMIT_MB = 1000
12
  YT_LENGTH_LIMIT_S = 3600 # limit to 1 hour YouTube files
 
1
  from transformers import pipeline, AutoModelForSpeechSeq2Seq, AutoProcessor
2
  from transformers.pipelines.audio_utils import ffmpeg_read
3
+ from huggingface_hub import login
4
  import yt_dlp as youtube_dl
5
  import gradio as gr
6
  import tempfile
 
8
  import time
9
  import os
10
 
11
+ login(os.environ["HF"], add_to_git_credential=True)
12
+
13
  BATCH_SIZE = 16
14
  FILE_LIMIT_MB = 1000
15
  YT_LENGTH_LIMIT_S = 3600 # limit to 1 hour YouTube files