amir22010 commited on
Commit
8df7942
·
1 Parent(s): f633cbf

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
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
- 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
 
 
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