Spaces:
Runtime error
Runtime error
VenkateshRoshan
commited on
Commit
·
713dd69
1
Parent(s):
dd44216
deployment file update
Browse files- .github/workflows/deploy.yml +3 -3
- src/deploy_sagemaker.py +1 -3
.github/workflows/deploy.yml
CHANGED
|
@@ -38,8 +38,8 @@ jobs:
|
|
| 38 |
|
| 39 |
- name: Build and push Docker image
|
| 40 |
run: |
|
| 41 |
-
docker build -t ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/
|
| 42 |
-
docker push ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/
|
| 43 |
|
| 44 |
- name: Deploy model to SageMaker
|
| 45 |
run: |
|
|
@@ -47,5 +47,5 @@ jobs:
|
|
| 47 |
--account_id ${{ secrets.ACCOUNT_ID }} \
|
| 48 |
--region ${{ secrets.AWS_REGION }} \
|
| 49 |
--role_arn ${{ secrets.SAGEMAKER_ROLE_ARN }} \
|
| 50 |
-
--ecr_repo_name "
|
| 51 |
--endpoint_name "customer-support-chatbot"
|
|
|
|
| 38 |
|
| 39 |
- name: Build and push Docker image
|
| 40 |
run: |
|
| 41 |
+
docker build -t ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/customer-support-chatbot:latest .
|
| 42 |
+
docker push ${{ secrets.ACCOUNT_ID }}.dkr.ecr.${{ secrets.AWS_REGION }}.amazonaws.com/customer-support-chatbot:latest
|
| 43 |
|
| 44 |
- name: Deploy model to SageMaker
|
| 45 |
run: |
|
|
|
|
| 47 |
--account_id ${{ secrets.ACCOUNT_ID }} \
|
| 48 |
--region ${{ secrets.AWS_REGION }} \
|
| 49 |
--role_arn ${{ secrets.SAGEMAKER_ROLE_ARN }} \
|
| 50 |
+
--ecr_repo_name "customer-support-chatbot" \
|
| 51 |
--endpoint_name "customer-support-chatbot"
|
src/deploy_sagemaker.py
CHANGED
|
@@ -26,14 +26,12 @@ def deploy_app(acc_id, region_name, role_arn, ecr_repo_name, endpoint_name="cust
|
|
| 26 |
|
| 27 |
# Define the image URI in ECR
|
| 28 |
ecr_image = f"{acc_id}.dkr.ecr.{region_name}.amazonaws.com/{ecr_repo_name}:latest"
|
| 29 |
-
model_name = f"customer-support-chatbot-{datetime.now().strftime('%Y%m%d-%H%M%S')}"
|
| 30 |
|
| 31 |
# Define model
|
| 32 |
model = Model(
|
| 33 |
image_uri=ecr_image,
|
| 34 |
role=role_arn,
|
| 35 |
-
sagemaker_session=sagemaker_session
|
| 36 |
-
model_name=model_name # Use the compliant model name
|
| 37 |
)
|
| 38 |
|
| 39 |
# Deploy model as a SageMaker endpoint
|
|
|
|
| 26 |
|
| 27 |
# Define the image URI in ECR
|
| 28 |
ecr_image = f"{acc_id}.dkr.ecr.{region_name}.amazonaws.com/{ecr_repo_name}:latest"
|
|
|
|
| 29 |
|
| 30 |
# Define model
|
| 31 |
model = Model(
|
| 32 |
image_uri=ecr_image,
|
| 33 |
role=role_arn,
|
| 34 |
+
sagemaker_session=sagemaker_session
|
|
|
|
| 35 |
)
|
| 36 |
|
| 37 |
# Deploy model as a SageMaker endpoint
|