new4u commited on
Commit
408571e
1 Parent(s): 8596cb6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -9,6 +9,11 @@ import tempfile
9
  import os
10
 
11
  MODEL_NAME = "openai/whisper-large-v3"
 
 
 
 
 
12
  device = 0 if torch.cuda.is_available() else "cpu"
13
  pipe = pipeline(
14
  task="automatic-speech-recognition",
 
9
  import os
10
 
11
  MODEL_NAME = "openai/whisper-large-v3"
12
+ BATCH_SIZE = 8
13
+ FILE_LIMIT_MB = 100000
14
+ YT_LENGTH_LIMIT_S = 360000 # limit to 1 hour YouTube files
15
+
16
+
17
  device = 0 if torch.cuda.is_available() else "cpu"
18
  pipe = pipeline(
19
  task="automatic-speech-recognition",