Rimi98 commited on
Commit
5561bb0
·
1 Parent(s): ad23ad1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,11 +19,11 @@ classes = ['Art', 'Astrology', 'Biology', 'Chemistry', 'Economics', 'History', '
19
  ### --- Audio/Video to txt ---###
20
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
21
  pipe = pipeline("automatic-speech-recognition",
22
- model="openai/whisper-base.en",
23
  chunk_length_s=30, device=device)
24
 
25
  ### --- Text Summary --- ###
26
- summarizer = pipeline("summarization", model="facebook/bart-large-cnn", device=device)
27
 
28
 
29
  def video_identity(video):
 
19
  ### --- Audio/Video to txt ---###
20
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
21
  pipe = pipeline("automatic-speech-recognition",
22
+ model="openai/whisper-tiny.en",
23
  chunk_length_s=30, device=device)
24
 
25
  ### --- Text Summary --- ###
26
+ summarizer = pipeline("summarization", model="facebook/bart-base", device=device)
27
 
28
 
29
  def video_identity(video):