docs: update bucket variable in geospatial-classification (GoogleCloudPlatform#7619)

* docs: update bucket variable in geospatial-classification

* Use the region variable

Co-authored-by: David Cavazos 

Author: parthea

people-and-planet-ai/geospatial-classification/README.ipynb

@@ -856,7 +856,7 @@
     "training_task = ee.batch.Export.table.toCloudStorage(\n",
     "    collection=ee.FeatureCollection(train_features),\n",
     "    description=\"Training image export\",\n",
-    "    bucket=bucket,\n",
+    "    bucket=cloud_storage_bucket,\n",
     "    fileNamePrefix=\"geospatial_training\",\n",
     "    selectors=BANDS + [LABEL],\n",
     "    fileFormat=\"TFRecord\",\n",
@@ -867,7 +867,7 @@
     "validation_task = ee.batch.Export.table.toCloudStorage(\n",
     "    collection=ee.FeatureCollection(validation_features),\n",
     "    description=\"Validation image export\",\n",
-    "    bucket=bucket,\n",
+    "    bucket=cloud_storage_bucket,\n",
     "    fileNamePrefix=\"geospatial_validation\",\n",
     "    selectors= BANDS + [LABEL],\n",
     "    fileFormat='TFRecord')\n",
@@ -928,7 +928,7 @@
    "source": [
     "from google.cloud import aiplatform\n",
     "\n",
-    "aiplatform.init(project=project, staging_bucket=bucket)\n",
+    "aiplatform.init(project=project, staging_bucket=cloud_storage_bucket)\n",
     "\n",
     "job = aiplatform.CustomTrainingJob(\n",
     "    display_name=\"geospatial_model_training\",\n",
@@ -955,7 +955,7 @@
    },
    "outputs": [],
    "source": [
-    "model = job.run(accelerator_type='NVIDIA_TESLA_K80', accelerator_count=1, args=[f'--bucket={bucket}'])"
+    "model = job.run(accelerator_type='NVIDIA_TESLA_K80', accelerator_count=1, args=[f'--bucket={cloud_storage_bucket}'])"
    ]
   },
   {
@@ -1013,7 +1013,7 @@
     "  --source=serving_app \\\n",
     "  --command=\"gunicorn\" \\\n",
     "  --args=\"--threads=8,--timeout=0,main:app\" \\\n",
-    "  --region=\"us-central1\" \\\n",
+    "  --region=\"{region}\" \\\n",
     "  --memory=\"1G\" \\\n",
     "  --no-allow-unauthenticated \\"
    ]
@@ -1246,7 +1246,7 @@
     "requests.post(\n",
     "    url=f'{service_url}/predict',\n",
     "    headers={'Authorization': f'Bearer {identity_token}'},\n",
-    "    json={'data': prediction_data, 'bucket': bucket},\n",
+    "    json={'data': prediction_data, 'bucket': cloud_storage_bucket},\n",
     ").json()['predictions']"
    ]
   },
@@ -1284,7 +1284,7 @@
     "    result = requests.post(\n",
     "      url=f'{service_url}/predict',\n",
     "      headers={'Authorization': f'Bearer {identity_token}'},\n",
-    "      json={'data': prediction_data, 'bucket': bucket},).json()\n",
+    "      json={'data': prediction_data, 'bucket': cloud_storage_bucket},).json()\n",
     "    return result['predictions']['predictions'][0][0][0][0]"
    ]
   },