From ee46587ab2c051d4ab462e53e9db36cd2ba92764 Mon Sep 17 00:00:00 2001
From: Francois Chollet <francois.chollet@gmail.com>
Date: Mon, 11 Dec 2023 20:09:24 -0800
Subject: [PATCH] Update requirements

---
 examples/vision/video_classification.py | 3 ++-
 requirements.txt                        | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/examples/vision/video_classification.py b/examples/vision/video_classification.py
index e5dc7c844a..17deea0905 100644
--- a/examples/vision/video_classification.py
+++ b/examples/vision/video_classification.py
@@ -218,7 +218,8 @@ def prepare_all_videos(df, root_dir):
             length = min(MAX_SEQ_LENGTH, video_length)
             for j in range(length):
                 temp_frame_features[i, j, :] = feature_extractor.predict(
-                    batch[None, j, :], verbose=0,
+                    batch[None, j, :],
+                    verbose=0,
                 )
             temp_frame_mask[i, :length] = 1  # 1 = not masked, 0 = masked
 
diff --git a/requirements.txt b/requirements.txt
index 72d975100c..cf855cbbcc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -10,7 +10,7 @@ pandas
 jupyter
 pycocotools
 pydot
-tensorflow==2.15.0
+tensorflow
 tensorflow_datasets
 keras-tuner==1.4.5
 keras-cv==0.7.1