Spaces:
Runtime error
Runtime error
VenkateshRoshan
commited on
Commit
·
1f83849
1
Parent(s):
9989061
instance_type updated
Browse files- src/deploy_sagemaker.py +1 -1
src/deploy_sagemaker.py
CHANGED
|
@@ -38,7 +38,7 @@ def deploy_app(acc_id, region_name, role_arn, ecr_repo_name, endpoint_name="cust
|
|
| 38 |
logger.info(f"Starting deployment of Gradio app to SageMaker endpoint {endpoint_name}...")
|
| 39 |
predictor = model.deploy(
|
| 40 |
initial_instance_count=1,
|
| 41 |
-
instance_type="ml.
|
| 42 |
endpoint_name=endpoint_name
|
| 43 |
)
|
| 44 |
logger.info(f"Gradio app deployed successfully to endpoint: {endpoint_name}")
|
|
|
|
| 38 |
logger.info(f"Starting deployment of Gradio app to SageMaker endpoint {endpoint_name}...")
|
| 39 |
predictor = model.deploy(
|
| 40 |
initial_instance_count=1,
|
| 41 |
+
instance_type="ml.g4dn.xlarge",
|
| 42 |
endpoint_name=endpoint_name
|
| 43 |
)
|
| 44 |
logger.info(f"Gradio app deployed successfully to endpoint: {endpoint_name}")
|