shreyasvaidya's picture
Upload folder using huggingface_hub
4c02d9f verified
import requests
# Define the translation function
def translate_text(source_text, source_lang="en", target_lang="hi"):
# API endpoint (replace with the actual endpoint if available)
api_url = "https://ai4bharat.iitm.ac.in/translate" # Placeholder URL
# Prepare payload and headers
payload = {
"service": "ai4bharat/indictrans--gpu-t4",
"source_language": source_lang,
"target_language": target_lang,
"text": source_text,
"enable_transliteration": True
}
headers = {
"Content-Type": "application/json"
}
# Send a POST request to the API
response = requests.post(api_url, json=payload, headers=headers)
# Handle the response
if response.status_code == 200:
translated_text = response.json().get("translated_text", "Translation failed")
return translated_text
else:
print(f"Error {response.status_code}: {response.text}")
return None
# Example usage
source_text = "Hello, how are you?"
source_language = "en" # English
target_language = "hi" # Hindi
translated_output = translate_text(source_text, source_language, target_language)
print(f"Translated Text: {translated_output}")