Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -21,10 +21,10 @@ model = AutoModelForSeq2SeqLM.from_pretrained("EzekielMW/LuoKslGloss")
|
|
21 |
|
22 |
# Where should output files be stored locally
|
23 |
# Where should output files be stored locally
|
24 |
-
drive_folder = "./
|
25 |
|
26 |
if not os.path.exists(drive_folder):
|
27 |
-
|
28 |
|
29 |
# Large batch sizes generally give good results for translation
|
30 |
effective_train_batch_size = 480
|
@@ -163,7 +163,7 @@ for code in LANGUAGE_CODES:
|
|
163 |
i = tokenizer.convert_tokens_to_ids(code_mapping[code])
|
164 |
tokenizer._added_tokens_encoder[code] = I
|
165 |
|
166 |
-
|
167 |
|
168 |
# Define a translation function
|
169 |
def translate(text, source_language, target_language):
|
|
|
21 |
|
22 |
# Where should output files be stored locally
|
23 |
# Where should output files be stored locally
|
24 |
+
drive_folder = "./quadserverlogs"
|
25 |
|
26 |
if not os.path.exists(drive_folder):
|
27 |
+
os.makedirs(drive_folder)
|
28 |
|
29 |
# Large batch sizes generally give good results for translation
|
30 |
effective_train_batch_size = 480
|
|
|
163 |
i = tokenizer.convert_tokens_to_ids(code_mapping[code])
|
164 |
tokenizer._added_tokens_encoder[code] = I
|
165 |
|
166 |
+
transformers.generation.utils.ForcedBOSTokenLogitsProcessor = transformers.ForcedBOSTokenLogitsProcessor
|
167 |
|
168 |
# Define a translation function
|
169 |
def translate(text, source_language, target_language):
|