Update app.py

app.py

@@ -17,6 +17,10 @@ def extract_video_frames(video_in):
     
     # Calculate the total duration of the video
     total_duration = clip.duration
+
+    if total_duration > 5:
+        clip = clip.subclip(0,5)
+        total_duration = clip.duration
     
     # Set the intervals to extract the frames
     intervals = [0, 2, 4, total_duration]
@@ -80,9 +84,12 @@ def infer(video_input, manual_caption, duration_in, seed, caption_output):
     
     # Load the input video file
     video_in = VideoFileClip(video_input)
+    duration = video_in.duration
+    if duration > 5:
+        video_in = video_in.subclip(0,5)
     
     # Make the audio the same length as the video
-    audio = audio.set_duration(video.duration)
+    audio = audio.set_duration(video_in.duration)
     
     # Combine the audio and video
     result = video_in.set_audio(audio)