import os
from huggingface_hub import InferenceClient

# 1. เชื่อมต่อกับ Hugging Face (ต้องมี HF_TOKEN)
client = InferenceClient(
    provider="hf-inference",
    api_key=os.environ.get("HF_TOKEN"), # ดึง Token จากเครื่องของคุณ
)

# 2. สั่งแปลภาษา โดยต้องระบุ src_lang และ tgt_lang เพิ่มเข้าไป
result = client.translation(
    "My name is Omar and I live in Zürich.",
    model="facebook/mbart-large-50-many-to-many-mmt",
    src_lang="en_XX", # ระบุภาษาต้นทาง (อังกฤษ)
    tgt_lang="de_DE"  # ระบุภาษาปลายทาง (เยอรมัน)
)

print(result)