Commit
·
f477097
1
Parent(s):
6b92718
small fixes
Browse files
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 |
|