xxx

app.py

@@ -1,10 +1,8 @@
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import streamlit as st
-from transformers import MarianMTModel, MarianTokenizer
 
-# Load the model and tokenizer
-model_name = "Helsinki-NLP/opus-mt-en-ro"
-tokenizer = MarianTokenizer.from_pretrained(model_name)
-model = MarianMTModel.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained("BlackKakapo/opus-mt-ro-en")
+model = AutoModelForSeq2SeqLM.from_pretrained("BlackKakapo/opus-mt-ro-en")
 
 # Streamlit app
 st.title("Romanian to English Translator")