Merge pull request #476 from satveerkhurpa/main

Removed static entries in the variables. Updated comments
aws-samples · Feb 4, 2025 · d160610 · d160610
2 parents 16019e2 + 8542bdc
commit d160610
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/responsible_ai/bedrock-guardrails/guardrails_custom_model_import.ipynb b/responsible_ai/bedrock-guardrails/guardrails_custom_model_import.ipynb
@@ -89,8 +89,8 @@
     "\n",
     "client = boto3.client(service_name = 'bedrock', region_name=region)\n",
     "\n",
-    "hf_model_id = \"deepseek-ai/DeepSeek-R1-Distill-Llama-8B\"  # The id of the model\n",
-    "model_id = \"arn:aws:bedrock:us-west-2:506556589049:imported-model/xpfqjbcs2bs1\"  #The ARN of the model imported using Custom Model Import\n",
+    "hf_model_id = \"<The id of the model>\"  # Replace the value with the id of the model\n",
+    "model_id = \"<arn of the model imported using Bedrock Custom Model Import>\"  # Replace the value with the ARN of the model imported using Custom Model Import\n",
     "\n",
     "# Enable hf_transfer for faster downloads\n",
     "os.environ[\"HF_HUB_ENABLE_HF_TRANSFER\"] = \"1\""