Update app.py

app.py

@@ -6,19 +6,21 @@ import tempfile
 import imageio
 from decord import VideoReader, cpu
 from transformers import pipeline
+from PIL import Image
 
 hf_token = os.environ.get("HUGGINGFACE_TOKEN")
 model_id = "google/gemma-3-27b-it"
 NUM_FRAMES = 8
 
 # 从视频中采样 N 帧
-def sample_video_frames(video_path, num_frames=NUM_FRAMES):
+def sample_video_frames(video_path, num_frames=8):
     vr = VideoReader(video_path, ctx=cpu(0))
     total_frames = len(vr)
     indices = [int(i) for i in torch.linspace(0, total_frames - 1, steps=num_frames)]
-    frames = [vr[i].asnumpy() for i in indices]
-    pil_frames = [imageio.core.util.Array(frame) for frame in frames]
-    return pil_frames
+    
+    # 关键点：强制转换为 PIL.Image
+    frames = [Image.fromarray(vr[i].asnumpy()) for i in indices]
+    return frames
 
 # 推理函数：加载模型、采样视频帧、推理
 @spaces.GPU