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 |
|