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
|