Update app.py

app.py

@@ -59,7 +59,6 @@ def get_subtitles_with_pytube(video_url):
     except Exception as e:
         return f"An error occurred while retrieving subtitles with pytube: {str(e)}"
 
-
 # Function to summarize YouTube video with selected model
 def summarize_youtube_video(url, model_name):
     transcript = get_youtube_transcript(url)
@@ -72,8 +71,11 @@ def summarize_youtube_video(url, model_name):
         transcript = transcript[:max_length]
 
     summarization_pipeline = create_summarization_pipeline(model_name)
-    summary = summarization_pipeline(transcript, min_length=10, max_length=1000, do_sample=False)
-    return summary[0]['summary_text']
+    try:
+        summary = summarization_pipeline(transcript, min_length=10, max_length=1000, do_sample=False)
+        return summary[0]['summary_text']
+    except Exception as e:
+        return f"An error occurred while summarizing: {str(e)}"
 
 # Define the Gradio interface
 iface = gr.Interface(
@@ -89,4 +91,3 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
-