Mhassanen commited on
Commit
2cbba07
·
verified ·
1 Parent(s): 6be851d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,9 +1,9 @@
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
- model_name2 = "Mhassanen/nllb-200-600M-En-Ar-finetuned"
5
- tokenizer = AutoTokenizer.from_pretrained(model_name2, src_lang="eng_Latn", tgt_lang="arz_Arab")
6
- model = AutoModelForSeq2SeqLM.from_pretrained(model_name2)
7
 
8
  def translate2(text):
9
  inputs = tokenizer(text, return_tensors="pt", padding=True)
 
1
  import streamlit as st
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
+ model_name = "Mhassanen/nllb-200-600M-En-Ar"
5
+ tokenizer = AutoTokenizer.from_pretrained(model_name, src_lang="eng_Latn", tgt_lang="arz_Arab")
6
+ model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
 
8
  def translate2(text):
9
  inputs = tokenizer(text, return_tensors="pt", padding=True)