revert code app.py as the approach to use GPU as needed doesnot work

app.py

@@ -11,6 +11,7 @@ from handlers import video_handler as vh
 model_path = "yolov8n.pt"  # YOLOv8 model path
 
 
+@spaces.GPU(duration=300)
 def process_video(video_file):
     """
     Processes the uploaded video file by extracting key frames, cropping them, and generating a processed video.
@@ -51,18 +52,19 @@ def process_video(video_file):
     # Step 1: Extract all frames
     status_message = "Extracting frames. Please wait...!"
     yield status_message, None
-    step_one(video_path, all_frames_folder, original_fps)
+    vh.extract_frames_by_rate(video_path, all_frames_folder, original_fps)
 
     # Step 2 - extract key frames
     status_message = "Extracting key frames. Please wait...!"
     yield status_message, None
-    step_two(all_frames_folder, key_frames_folder, original_fps)
+    fh.extract_key_frames(all_frames_folder, key_frames_folder, original_fps, model_path)
 
     # Step 3 - cropping key frames while reserving key objects
     status_message = "Cropping key frames. Please wait...!"
     yield status_message, None
     target_resolution = (360, 640)  # Output resolution (9:16)
-    step_three(key_frames_folder, cropped_frames_folder, target_resolution)
+    fh.crop_preserve_key_objects(key_frames_folder, cropped_frames_folder, model_path, target_resolution)
+
 
     status_message = "Generating final video. Please wait...!"
     yield status_message, None
@@ -74,21 +76,6 @@ def process_video(video_file):
     status_message = "Processing complete!"
     yield status_message, processed_video_path
 
-
-def step_one(video_path, all_frames_folder, original_fps):
-    vh.extract_frames_by_rate(video_path, all_frames_folder, original_fps)
-
-
-@spaces.GPU(duration=300)
-def step_two(all_frames_folder, key_frames_folder, original_fps):
-    fh.extract_key_frames(all_frames_folder, key_frames_folder, original_fps, model_path)
-
-
-@spaces.GPU(duration=120)
-def step_three(key_frames_folder, cropped_frames_folder, target_resolution):
-    fh.crop_preserve_key_objects(key_frames_folder, cropped_frames_folder, model_path, target_resolution)
-
-
 # Gradio Blocks UI
 with gr.Blocks() as demo:
     gr.Markdown("## Generate short video for your football match")