Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -111,8 +111,8 @@ def generate_sentence_embeddings(sentences, model, pattern):
|
|
| 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(
|
| 115 |
-
model = MarianMTModel.from_pretrained(
|
| 116 |
return tokenizer, model
|
| 117 |
|
| 118 |
# Function to perform translation
|
|
|
|
| 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
|