Update app.py

app.py

@@ -116,9 +116,9 @@ def process_video_for_canny(video, width, height):
     canny_video = []
     detect_resolution = max(video[0].size[0],video[0].size[1])
     image_resolution = max(width, height)
-    
-    for frame in video: 
-        canny_video.append(canny_processor(frame, low_threshold=50, high_threshold=200, detect_resolution=detect_resolution, image_resolution=image_resolution))
+    with imageio.get_reader(video) as reader:
+        for frame in reader: 
+            canny_video.append(canny_processor(frame, low_threshold=50, high_threshold=200, detect_resolution=detect_resolution, image_resolution=image_resolution))
      
     return canny_video
 
@@ -132,7 +132,7 @@ def process_input_video(reference_video, width, height,
     
     try:
         # Load video into a list of PIL images
-        video = load_video(reference_video)
+        #video = load_video(reference_video)
         
         # Process video for canny edges
         processed_video = process_video_for_canny(video, width, height)
@@ -150,7 +150,7 @@ def process_input_video(reference_video, width, height,
 
 def process_video_for_control(reference_video, control_type, width, height):
     """Process video based on the selected control type - now only used for non-canny types"""
-    video = load_video(reference_video)
+    # video = load_video(reference_video)
     
     if control_type == "canny":
         # This should not be called for canny since it's pre-processed