IinjyI commited on
Commit
174b69d
·
verified ·
1 Parent(s): 1a12845

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ nltk.download('stopwords')
19
 
20
  model_checkpoint = "marefa-nlp/marefa-mt-en-ar"
21
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
22
- model = TFAutoModelForSeq2SeqLM.from_pretrained("tf_model/")
23
 
24
  # Load cleaned_word_embeddings
25
  with open("cleaned_word_embeddings.pkl", "rb") as f:
 
19
 
20
  model_checkpoint = "marefa-nlp/marefa-mt-en-ar"
21
  tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
22
+ model = TFAutoModelForSeq2SeqLM.from_pretrained(model_checkpoint,from_pt=True)
23
 
24
  # Load cleaned_word_embeddings
25
  with open("cleaned_word_embeddings.pkl", "rb") as f: