Spaces:
Running
Running
kovacsvi
commited on
Commit
·
ebe8358
1
Parent(s):
eb87e48
debug (domain list)
Browse files- interfaces/cap.py +1 -0
interfaces/cap.py
CHANGED
|
@@ -105,6 +105,7 @@ def predict(text, model_id, tokenizer_id):
|
|
| 105 |
return output_pred, output_info
|
| 106 |
|
| 107 |
def predict_cap(text, language, domain):
|
|
|
|
| 108 |
domain = domains[domain]
|
| 109 |
model_id = build_huggingface_path(language, domain)
|
| 110 |
tokenizer_id = "xlm-roberta-large"
|
|
|
|
| 105 |
return output_pred, output_info
|
| 106 |
|
| 107 |
def predict_cap(text, language, domain):
|
| 108 |
+
print(domain) # debug statement
|
| 109 |
domain = domains[domain]
|
| 110 |
model_id = build_huggingface_path(language, domain)
|
| 111 |
tokenizer_id = "xlm-roberta-large"
|