Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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-
|
23 |
chunk_length_s=30, device=device)
|
24 |
|
25 |
### --- Text Summary --- ###
|
26 |
-
summarizer = pipeline("summarization", model="facebook/bart-
|
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):
|