Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -15,6 +15,7 @@ whisper_model = whisper.load_model("tiny")
|
|
15 |
|
16 |
# Load the text summarization model from Hugging Face
|
17 |
summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
|
|
|
18 |
|
19 |
# Function to transcribe and summarize the audio file
|
20 |
def transcribe_and_summarize(audio):
|
@@ -24,6 +25,7 @@ def transcribe_and_summarize(audio):
|
|
24 |
|
25 |
# Step 2: Summarize the transcription
|
26 |
summary = summarizer(transcription, min_length=50, max_length=1000)
|
|
|
27 |
summary_text = summary[0]['summary_text']
|
28 |
|
29 |
return transcription, summary_text
|
|
|
15 |
|
16 |
# Load the text summarization model from Hugging Face
|
17 |
summarizer = pipeline(task="summarization", model="facebook/bart-large-cnn")
|
18 |
+
summarizer = pipeline("summarization", model="google/pegasus-xsum")
|
19 |
|
20 |
# Function to transcribe and summarize the audio file
|
21 |
def transcribe_and_summarize(audio):
|
|
|
25 |
|
26 |
# Step 2: Summarize the transcription
|
27 |
summary = summarizer(transcription, min_length=50, max_length=1000)
|
28 |
+
|
29 |
summary_text = summary[0]['summary_text']
|
30 |
|
31 |
return transcription, summary_text
|