diff --git a/main.py b/main.py index a4f70d7..711f70f 100644 --- a/main.py +++ b/main.py @@ -58,4 +58,4 @@ def rec_score(first_id: str, second_id: str): emotion_diff = 1 - cosine(first['emotions'], second['emotions']) FS.set_variable('EMOTIONS', emotion_diff) - return {'score': FS.inference(['RECOMMENDATION'])} + return FS.inference(['RECOMMENDATION'])