Walid-Ahmed commited on
Commit
863d23e
·
verified ·
1 Parent(s): 3b63114

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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