Update app.py

app.py

@@ -191,23 +191,22 @@ def extract_frames(video_path, num_frames):
 
 # Gradio app logic
 def process_video(video, num_frames):
-    print(video)
-    video_path = os.path.join(tempfile.gettempdir(), video)
-    with open(video_path, "wb") as f:
-        f.write(video.read())
     
-    frames = extract_frames(video_path, num_frames)
+    # Extract frames from the video
+    frames = extract_frames(video, num_frames)
     
     nsfw_count = 0
     total_frames = len(frames)
 
+    # Check each frame for NSFW content
     for frame_path in frames:
         if check_nsfw_final(frame_path):
             nsfw_count += 1
-        os.remove(frame_path) 
+        os.remove(frame_path)  # Clean up temp frame files
 
     return f"{nsfw_count} out of {total_frames} frames were NSFW."
 
+
 with gr.Blocks() as app:
     gr.Markdown("## NSFW Frame Detection in Video")