razhan commited on
Commit
a449bcc
1 Parent(s): fbe308b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -5,10 +5,14 @@ import yt_dlp as youtube_dl
5
 
6
  from transformers import pipeline
7
  from huggingface_hub import model_info
 
 
 
8
 
9
  MODEL_NAME = "razhan/whisper-small-ckb"
10
  BATCH_SIZE = 1
11
  FILE_LIMIT_MB = 10
 
12
 
13
  device = 0 if torch.cuda.is_available() else "cpu"
14
  pipe = pipeline(
 
5
 
6
  from transformers import pipeline
7
  from huggingface_hub import model_info
8
+ import re
9
+ import tempfile
10
+ import os
11
 
12
  MODEL_NAME = "razhan/whisper-small-ckb"
13
  BATCH_SIZE = 1
14
  FILE_LIMIT_MB = 10
15
+ YT_LENGTH_LIMIT_S = 60 * 10
16
 
17
  device = 0 if torch.cuda.is_available() else "cpu"
18
  pipe = pipeline(