clear repo

Author: pirocheto

test_model/.gitignore

@@ -0,0 +1,13 @@
+# Python-generated files
+__pycache__/
+*.py[oc]
+build/
+dist/
+wheels/
+*.egg-info
+
+# Virtual environments
+.venv
+
+.pytest_cache/
+.mypy_cache/

test_model/.python-version

@@ -0,0 +1 @@
+3.12

test_model/pyproject.toml

@@ -0,0 +1,28 @@
+[project]
+name = "aws-sagemaker-hf-model-deployment"
+version = "0.1.0"
+description = "AWS SageMaker Hugging Face Model Deployment"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = ["boto3>=1.35.97"]
+
+[tool.mypy]
+disallow_untyped_calls = true
+
+[tool.ruff]
+target-version = "py312"
+
+[tool.ruff.lint]
+select = [
+    "F",    # Pyflakes
+    "I",    # isort
+    "E",    # pycodestyle
+    "S",    # flake8-bandit
+    "N",    # pep8-naming
+    "PERF",
+]
+
+[dependency-groups]
+dev = [
+    "boto3-stubs>=1.36.1",
+]

test_model/test_endpoint.py

@@ -0,0 +1,106 @@
+import json
+import time
+
+import boto3
+
+AWS_REGION = "us-east-1"
+ENDPOINT_NAME = "prod-multilingual-sentiment-analysis-endpoint"
+
+sagemaker_runtime = boto3.client("sagemaker-runtime", region_name=AWS_REGION)
+
+texts = [
+    # English
+    "I absolutely love the new design of this app!",
+    "The customer service was disappointing.",
+    "The weather is fine, nothing special.",
+    # Chinese
+    "这家餐厅的菜味道非常棒！",
+    "我对他的回答很失望。",
+    "天气今天一般。",
+    # Spanish
+    "¡Me encanta cómo quedó la decoración!",
+    "El servicio fue terrible y muy lento.",
+    "El libro estuvo más o menos.",
+    # Arabic
+    "الخدمة في هذا الفندق رائعة جدًا!",
+    "لم يعجبني الطعام في هذا المطعم.",
+    "كانت الرحلة عادية。",
+    # Ukrainian
+    "Мені дуже сподобалася ця вистава!",
+    "Обслуговування було жахливим.",
+    "Книга була посередньою。",
+    # Hindi
+    "यह जगह सच में अद्भुत है!",
+    "यह अनुभव बहुत खराब था।",
+    "फिल्म ठीक-ठाक थी।",
+    # Bengali
+    "এখানকার পরিবেশ অসাধারণ!",
+    "সেবার মান একেবারেই খারাপ।",
+    "খাবারটা মোটামুটি ছিল।",
+    # Portuguese
+    "Este livro é fantástico! Eu aprendi muitas coisas novas e inspiradoras.",
+    "Não gostei do produto, veio quebrado.",
+    "O filme foi ok, nada de especial.",
+    # Japanese
+    "このレストランの料理は本当に美味しいです！",
+    "このホテルのサービスはがっかりしました。",
+    "天気はまあまあです。",
+    # Russian
+    "Я в восторге от этого нового гаджета!",
+    "Этот сервис оставил у меня только разочарование.",
+    "Встреча была обычной, ничего особенного.",
+    # French
+    "J'adore ce restaurant, c'est excellent !",
+    "L'attente était trop longue et frustrante.",
+    "Le film était moyen, sans plus.",
+    # Turkish
+    "Bu otelin manzarasına bayıldım!",
+    "Ürün tam bir hayal kırıklığıydı.",
+    "Konser fena değildi, ortalamaydı.",
+    # Italian
+    "Adoro questo posto, è fantastico!",
+    "Il servizio clienti è stato pessimo.",
+    "La cena era nella media.",
+    # Polish
+    "Uwielbiam tę restaurację, jedzenie jest świetne!",
+    "Obsługa klienta była rozczarowująca.",
+    "Pogoda jest w porządku, nic szczególnego.",
+    # Tagalog
+    "Ang ganda ng lugar na ito, sobrang aliwalas!",
+    "Hindi maganda ang serbisyo nila dito.",
+    "Maayos lang ang palabas, walang espesyal.",
+    # Dutch
+    "Ik ben echt blij met mijn nieuwe aankoop!",
+    "De klantenservice was echt slecht.",
+    "De presentatie was gewoon oké, niet bijzonder.",
+    # Malay
+    "Saya suka makanan di sini, sangat sedap!",
+    "Pengalaman ini sangat mengecewakan.",
+    "Hari ini cuacanya biasa sahaja.",
+    # Korean
+    "이 가게의 케이크는 정말 맛있어요!",
+    "서비스가 너무 별로였어요.",
+    "날씨가 그저 그렇네요.",
+    # Swiss German
+    "Ich find dä Service i de Beiz mega guet!",
+    "Däs Esä het mir nöd gfalle.",
+    "D Wätter hüt isch so naja.",
+]
+
+start = time.time()
+
+response = sagemaker_runtime.invoke_endpoint(
+    EndpointName=ENDPOINT_NAME,
+    ContentType="application/json",
+    Body=json.dumps({"inputs": texts}),
+)
+
+duration = time.time() - start
+
+result = json.loads(response["Body"].read())
+for text, sentiment in zip(texts, result):
+    print(f"Text: {text}")
+    print(f"Sentiment: {sentiment}")
+    print()
+
+print(f"Duration: {duration:.2f}s for {len(texts)} texts")