File size: 1,236 Bytes
c95e994 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | from googletrans import Translator
import torch
from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
def translate_to_english(text):
translator = Translator()
translated_text = translator.translate(text, src='kn', dest='en')
return translated_text.text
def main():
# Input text in Kannada
kannada_text = input("Enter the text you want to translate to English: ")
# Translate Kannada text to English
english_text = translate_to_english(kannada_text)
print("Translated text in English:", english_text)
# Load DistilBERT model and tokenizer
tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
model = DistilBertForSequenceClassification.from_pretrained("distilbert-base-uncased-finetuned-sst-2-english")
# Tokenize the English text
inputs = tokenizer(english_text, return_tensors="pt")
# Classify the text
with torch.no_grad():
logits = model(**inputs).logits
# Get the predicted class
predicted_class_id = logits.argmax().item()
predicted_label = model.config.id2label[predicted_class_id]
print("Predicted class:", predicted_label)
if __name__ == "__main__":
main()
|