Updated app.py

app.py

@@ -3,11 +3,9 @@ from pytube import YouTube
 from moviepy.editor import *
 import torch
 from transformers import Wav2Vec2ForCTC, Wav2Vec2CTCTokenizer, pipeline, logging
-from transformers import pipeline
 import librosa
 
-
-# Suppress warnings from transformers
+# Suppress warnings from transformers to clean up the output
 logging.set_verbosity_error()
 
 # Streamlit interface setup
@@ -48,22 +46,26 @@ if st.button('Summarize'):
                 predicted_ids = torch.argmax(logits, dim=-1)
 
                 # Decode the speech
-                transcription = tokenizer.decode(predicted_ids[0])
+                transcription = tokenizer.batch_decode(predicted_ids, skip_special_tokens=True)[0]  # Ensure transcription is a single string
 
-                # Show progress
-                st.progress(75)
+                # Ensure transcription is a string and not empty
+                if isinstance(transcription, str) and transcription.strip():
+                    # Show progress
+                    st.progress(75)
 
-                # Summarization
-                summarizer = pipeline("summarization")
-                summary = summarizer(transcription, max_length=130, min_length=30, do_sample=False)
+                    # Summarization
+                    summarizer = pipeline("summarization")
+                    summary = summarizer(transcription, max_length=130, min_length=30, do_sample=False)
 
-                # Display the summary
-                st.success("Done!")
-                st.write("### Summary:")
-                st.write(summary[0]['summary_text'])
+                    # Display the summary
+                    st.success("Done!")
+                    st.write("### Summary:")
+                    st.write(summary[0]['summary_text'])
 
-                # Final progress
-                st.progress(100)
+                    # Final progress
+                    st.progress(100)
+                else:
+                    st.error("Could not transcribe audio or transcription is empty.")
 
             except Exception as e:
                 st.error(f"An error occurred: {e}")