Update app.py (#7)

- Update app.py (4b9bca0dab3c9feb9c3e6bb59106709a35b9313d)


Co-authored-by: Linoy Tsaban <linoyts@users.noreply.huggingface.co>

app.py

@@ -117,7 +117,7 @@ def process_video_for_canny(video, width, height):
     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))
+        canny_video.append(canny_processor(frame, low_threshold=50, high_threshold=200, detect_resolution=detect_resolution, image_resolution=image_resolution, output_type="np"))
      
     return canny_video
 
@@ -137,11 +137,14 @@ def process_input_video(reference_video, width, height,
         processed_video = process_video_for_canny(video, width, height)
         
         # Create a preview video file for display
-        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmp_file:
-            preview_path = tmp_file.name
-            export_to_video(processed_video, preview_path, fps=FPS)
+        output_filename = "control.mp4"
+        video_uint8_frames = [(frame * 255).astype(np.uint8) for frame in processed_video]
+        with imageio.get_writer(output_filename, fps=FPS, quality=8, macro_block_size=1) as writer: 
+             for frame_idx, frame_data in enumerate(video_uint8_frames):
+                progress((frame_idx + 1) / len(video_uint8_frames), desc="Encoding video frames...")
+                writer.append_data(frame_data)
         
-        return preview_path
+        return output_filename
         
     except Exception as e:
         print(f"Error processing input video: {e}")