csukuangfj commited on
Commit
f477097
·
1 Parent(s): 6b92718

small fixes

Browse files
Files changed (1) hide show
  1. model.py +4 -0
model.py CHANGED
@@ -234,6 +234,10 @@ def get_pretrained_model(
234
  return vietnamese_models[repo_id](
235
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
236
  )
 
 
 
 
237
  else:
238
  raise ValueError(f"Unsupported repo_id: {repo_id}")
239
 
 
234
  return vietnamese_models[repo_id](
235
  repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
236
  )
237
+ elif repo_id in portuguese_brazlian_models:
238
+ return portuguese_brazlian_models[repo_id](
239
+ repo_id, decoding_method=decoding_method, num_active_paths=num_active_paths
240
+ )
241
  else:
242
  raise ValueError(f"Unsupported repo_id: {repo_id}")
243