Integ test updates

Author: samruds

src/sagemaker/serve/builder/tei_builder.py

@@ -16,10 +16,11 @@
 from typing import Type
 from abc import ABC, abstractmethod
 
+from sagemaker import image_uris
 from sagemaker.model import Model
 from sagemaker.djl_inference.model import _get_model_config_properties_from_hf
 
-from sagemaker.huggingface import HuggingFaceModel, get_huggingface_llm_image_uri
+from sagemaker.huggingface import HuggingFaceModel
 from sagemaker.serve.utils.local_hardware import (
     _get_nb_instance,
 )
@@ -87,8 +88,10 @@ def _set_to_tgi(self):
     def _create_tei_model(self) -> Type[Model]:
         """Placeholder docstring"""
         if not self.image_uri:
-            self.image_uri = get_huggingface_llm_image_uri(
-                "huggingface-tei", session=self.sagemaker_session
+            self.image_uri = image_uris.retrieve(
+                "huggingface-tei",
+                image_scope="inference",
+                instance_type=self.nb_instance_type
             )
 
         pysdk_model = HuggingFaceModel(

src/sagemaker/serve/model_server/tgi/server.py

@@ -74,7 +74,7 @@ def _invoke_tgi_serving(self, request: object, content_type: str, accept: str):
         """Placeholder docstring"""
         try:
             response = requests.post(
-                f"http://{get_docker_host()}:8080/generate",
+                f"http://{get_docker_host()}:8080/invocations",
                 data=request,
                 headers={"Content-Type": content_type, "Accept": accept},
                 timeout=600,