Spaces:
Build error
Build error
Fix format issue
Browse files- language_model.py +8 -5
language_model.py
CHANGED
|
@@ -165,10 +165,13 @@ class TransformersAVG:
|
|
| 165 |
def __format_av(self, predictions: str, splitting_symbol: str = '|') -> list:
|
| 166 |
my_list = []
|
| 167 |
for raw_string in predictions.split(splitting_symbol):
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
|
|
|
|
|
|
|
|
|
| 173 |
|
| 174 |
return my_list
|
|
|
|
| 165 |
def __format_av(self, predictions: str, splitting_symbol: str = '|') -> list:
|
| 166 |
my_list = []
|
| 167 |
for raw_string in predictions.split(splitting_symbol):
|
| 168 |
+
try:
|
| 169 |
+
a = raw_string.replace('attribute: ', '')
|
| 170 |
+
a = a.split(',')[0].strip()
|
| 171 |
+
v = raw_string.replace('value: ', '')
|
| 172 |
+
v = v.split(',')[1].strip()
|
| 173 |
+
my_list.append((a,v))
|
| 174 |
+
except:
|
| 175 |
+
pass
|
| 176 |
|
| 177 |
return my_list
|