Updated `setup.sh`. Upgrade `keras` and `tensorflow`

setup.sh

@@ -11,5 +11,8 @@ python -m pip install --upgrade pip
 echo "Installing requirements..."
 pip install -r requirements.txt
 
+echo "Upgrading TensorFlow and Keras..."
+pip install --upgrade tensorflow keras
+
 echo "Setup complete! To activate the environment, run:"
 echo "source tf_test_env/bin/activate"