Spaces:
Sleeping
Sleeping
print result of phoBert
Browse files- phoBERT.py +1 -0
phoBERT.py
CHANGED
@@ -78,6 +78,7 @@ def BERT_predict(text):
|
|
78 |
token_type_ids = token['token_type_ids']
|
79 |
|
80 |
result = model(ids, mask, token_type_ids)
|
|
|
81 |
# print(result)
|
82 |
return result.tolist()[0]
|
83 |
|
|
|
78 |
token_type_ids = token['token_type_ids']
|
79 |
|
80 |
result = model(ids, mask, token_type_ids)
|
81 |
+
print(result)
|
82 |
# print(result)
|
83 |
return result.tolist()[0]
|
84 |
|