from sagemaker.jumpstart.model import JumpStartModel
import json

model_id = "huggingface-sentencesimilarity-bge-small-en-v1-5"
endpoint_input = json.dumps(["How cute your dog is!", "Your dog is so cute."])

model = JumpStartModel(model_id=model_id)
predictor = model.deploy()
response = predictor.predict(endpoint_input)
print(f"Inference:\nInput: {endpoint_input}\nResponse: {response}\n")