Blessmore commited on
Commit
3146cd6
·
verified ·
1 Parent(s): fa6d672

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -110,9 +110,9 @@ def generate_sentence_embeddings(sentences, model, pattern):
110
 
111
  # Function to load the translation model and tokenizer
112
  def load_translation_model(model_folder):
113
- model_path = os.path.join(model_folder, "shona_to_english_model")
114
- tokenizer = MarianTokenizer.from_pretrained(model_folder)
115
- model = MarianMTModel.from_pretrained(model_folder)
116
  return tokenizer, model
117
 
118
  # Function to perform translation
 
110
 
111
  # Function to load the translation model and tokenizer
112
  def load_translation_model(model_folder):
113
+ model_path = os.path.join(model_folder)
114
+ tokenizer = MarianTokenizer.from_pretrained(model_path)
115
+ model = MarianMTModel.from_pretrained(model_path)
116
  return tokenizer, model
117
 
118
  # Function to perform translation