Upload app.py

app.py

@@ -114,10 +114,12 @@ def simulate(image0, image1):
     avg_prob = sum(probs) / len(preferences)
 
     preference = 0 if avg_preference < 0.5 else 1
-    
-    right_reasons = [reasons[i] for i in range(len(reasons)) if preferences[i] == preference]
-    reasons_list = ast.literal_eval(right_reasons)
-    all_reasons = ','.join(reasons_list)
+
+    all_reasons = []
+    for item in reasons:
+        words = [word.strip() for word in item.split(',')]
+        all_reasons.extend(words)
+
     wordcloud = WordCloud(width = 800, height = 400, background_color ='white').generate(all_reasons)
 
     plt.figure(figsize=(10,5))