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