File tree Expand file tree Collapse file tree 1 file changed +1
-6
lines changed
Expand file tree Collapse file tree 1 file changed +1
-6
lines changed Original file line number Diff line number Diff line change @@ -282,12 +282,7 @@ def setup_environment(self):
282282 self .bucket = f"{ self .bucket_basename } -{ self .region } "
283283
284284 # Set UDOP model path based on region
285- if self .region == "us-east-1" :
286- self .public_sample_udop_model = "s3://aws-ml-blog-us-east-1/artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
287- elif self .region == "us-west-2" :
288- self .public_sample_udop_model = "s3://aws-ml-blog-us-west-2/artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
289- else :
290- self .public_sample_udop_model = ""
285+ self .public_sample_udop_model = f"s3://aws-ml-blog-{ self .region } /artifacts/genai-idp/udop-finetuning/rvl-cdip/model.tar.gz"
291286
292287 def check_prerequisites (self ):
293288 """Check for required commands and versions"""
You can’t perform that action at this time.
0 commit comments