Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import streamlit as st
 import cv2
 import mediapipe as mp
@@ -46,27 +45,27 @@ class VideoProcessor:
         self.model = build_model()
     
     def process_video(self, video_file):
-    # Get the filename from the file object
-    filename = video_file.name
-    # Create a temporary file to write the contents of the uploaded video file
-    temp_file = open(filename, 'wb')
-    temp_file.write(video_file.read())
-    temp_file.close()
-    # Now we can open the video file using cv2.VideoCapture()
-    cap = cv2.VideoCapture(filename)
-    out_frames = []
-    while cap.isOpened():
-        ret, frame = cap.read()
-        if not ret:
-            break
-        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        results = self.pose.process(frame_rgb)
-        frame = self.draw_landmarks(frame, results)
-        out_frames.append(frame)
-    cap.release()
-    # Remove the temporary file
-    os.remove(filename)
-    return out_frames
+        # Get the filename from the file object
+        filename = video_file.name
+        # Create a temporary file to write the contents of the uploaded video file
+        temp_file = open(filename, 'wb')
+        temp_file.write(video_file.read())
+        temp_file.close()
+        # Now we can open the video file using cv2.VideoCapture()
+        cap = cv2.VideoCapture(filename)
+        out_frames = []
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            results = self.pose.process(frame_rgb)
+            frame = self.draw_landmarks(frame, results)
+            out_frames.append(frame)
+        cap.release()
+        # Remove the temporary file
+        os.remove(filename)
+        return out_frames
     
     def draw_landmarks(self, image, results):
         mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_pose.POSE_CONNECTIONS,