EzekielMW commited on
Commit
9b32dc4
·
verified ·
1 Parent(s): ef1bd65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 = "./serverlogs"
25
 
26
  if not os.path.exists(drive_folder):
27
- %mkdir $drive_folder
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
- #transformers.generation.utils.ForcedBOSTokenLogitsProcessor = transformers.ForcedBOSTokenLogitsProcessor
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):