Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
app.py
CHANGED
|
@@ -65,15 +65,15 @@ def perform_asde_inference(text, dataset, model_id):
|
|
| 65 |
output = double_keybert_generator.generate(tokenized_text.input_ids,max_length=512)
|
| 66 |
result = tokenizer_keybert.decode(output[0], skip_special_tokens=True)
|
| 67 |
'''
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 73 |
-
|
| 74 |
-
|
| 75 |
-
|
| 76 |
-
|
| 77 |
|
| 78 |
return pred_doubles, true_doubles, text, model_generated
|
| 79 |
|
|
|
|
| 65 |
output = double_keybert_generator.generate(tokenized_text.input_ids,max_length=512)
|
| 66 |
result = tokenizer_keybert.decode(output[0], skip_special_tokens=True)
|
| 67 |
'''
|
| 68 |
+
pred_asp = [i.split(':')[0] for i in model_generated.split(',')]
|
| 69 |
+
pred_sent = [i.split(':')[1] for i in model_generated.split(',')]
|
| 70 |
+
|
| 71 |
+
pred_doubles = pd.DataFrame(list(map(list, zip(pred_asp, pred_sent))),columns=['Aspect','Sentiment'])
|
| 72 |
+
|
| 73 |
+
if not text:
|
| 74 |
+
true_doubles = pd.DataFrame(list(map(list, zip(ast.literal_eval(true_aspect), ast.literal_eval(true_sentiment)))),columns=['Aspect','Sentiment'])
|
| 75 |
+
else:
|
| 76 |
+
true_doubles = pd.DataFrame()
|
| 77 |
|
| 78 |
return pred_doubles, true_doubles, text, model_generated
|
| 79 |
|