Spaces:
Build error
Build error
Update app.py
Browse files
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
|
| 114 |
-
tokenizer = MarianTokenizer.from_pretrained(
|
| 115 |
-
model = MarianMTModel.from_pretrained(
|
| 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
|