tudorgeorgescu commited on
Commit
700b352
·
verified ·
1 Parent(s): fe817f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,8 +1,8 @@
1
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
  import streamlit as st
3
 
4
- tokenizer = AutoTokenizer.from_pretrained("BlackKakapo/opus-mt-ro-en")
5
- model = AutoModelForSeq2SeqLM.from_pretrained("BlackKakapo/opus-mt-ro-en")
6
 
7
  # Streamlit app
8
  st.title("Romanian to English Translator")
 
1
+ from transformers import MarianTokenizer, MarianMTModel
2
  import streamlit as st
3
 
4
+ tokenizer = MarianTokenizer.from_pretrained("BlackKakapo/opus-mt-ro-en")
5
+ model = MarianMTModel.from_pretrained("BlackKakapo/opus-mt-ro-en")
6
 
7
  # Streamlit app
8
  st.title("Romanian to English Translator")