Update app.py

app.py

@@ -1,5 +1,5 @@
+import spaces
 import gradio as gr
-from gradio import spaces
 import cv2
 from PIL import Image, ImageDraw, ImageFont
 import torch
@@ -46,7 +46,7 @@ def process_video(video_path, target, progress=gr.Progress()):
     temp_dir = tempfile.mkdtemp()
     frame_paths = []
 
-    batch_size = 4  # Process 4 frames at a time
+    batch_size = 2  # Process 2 frames at a time
     batch_frames = []
     batch_indices = []
 
@@ -57,8 +57,7 @@ def process_video(video_path, target, progress=gr.Progress()):
         if not ret:
             break
 
-        # Resize the frame
-        #img_resized = cv2.resize(img, (640, 360))
+        # Convert to RGB without resizing
         pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
         
         batch_frames.append(pil_img)