translation_model.py

import time
from transformers import pipeline

# Load the translation model at the start to avoid reloading on each request
start_time = time.time()

pipe = pipeline("text2text-generation", model="sarch7040/praTranv2")
print(f"Model loaded in {time.time() - start_time} seconds")

def translate(input_text):
    print(f"Input text: {input_text}")
    output_text = pipe(
        input_text,
        do_sample=True,
        num_beams=5,
        max_length=50,
        no_repeat_ngram_size=2,
        temperature=0.7
    )
    translated_text = output_text[0]['generated_text']
    print(f"Output text: {translated_text}")
    return translated_text