Update app.py

app.py

@@ -43,6 +43,12 @@ def summarize_youtube_video(url, model_name):
     transcript = get_youtube_transcript(url)
     if "An error occurred" in transcript:
         return transcript
+
+    # Truncate the transcript if necessary
+    max_length = 1024  # Adjust according to the model's maximum sequence length
+    if len(transcript) > max_length:
+        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']
@@ -61,3 +67,4 @@ iface = gr.Interface(
 
 if __name__ == "__main__":
     iface.launch()
+