File size: 464 Bytes
5607d73
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
from langdetect import detect
from transformers import pipeline

# Detect language of the input text
def detect_language(text: str) -> str:
    try:
        return detect(text)
    except:
        return "unknown"

# Translate text to English (or another target language)
def translate_text(text: str, target_lang: str = "en") -> str:
    translator = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
    return translator(text)[0]["translation_text"]