from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "srinidhi05/K2E-Translator"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(model_name)
def translate_korean(text):
prompt = f"Translate Korean to English: {text} ->"
inputs = tokenizer(prompt, return_tensors="pt")
output = model.generate(**inputs, max_length=100)
return tokenizer.decode(output[0], skip_special_tokens=True)
print(translate_korean("오늘 날씨가 어때요?")) # Expected output: "How's the weather today?"
