x

app.py

@@ -1,8 +1,10 @@
 import streamlit as st
-from transformers import pipeline
+from transformers import MarianMTModel, MarianTokenizer
 
-# Initialize the translation pipeline
-translator = pipeline("translation_ro_to_en")
+# Load the model and tokenizer
+model_name = "Helsinki-NLP/opus-mt-ro-en"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
 
 # Streamlit app
 st.title("Romanian to English Translator")
@@ -13,8 +15,11 @@ input_text = st.text_area("Enter text in Romanian:")
 # Translate button
 if st.button("Translate"):
     if input_text:
+        # Tokenize the input text
+        inputs = tokenizer(input_text, return_tensors="pt", padding=True)
         # Perform translation
-        translation = translator(input_text)[0]['translation_text']
+        translated_tokens = model.generate(**inputs)
+        translation = tokenizer.decode(translated_tokens[0], skip_special_tokens=True)
         st.write("Translation:")
         st.write(translation)
     else: