Update app.py

app.py

@@ -106,9 +106,9 @@ def retrive_data(temp_data, img_key, mode='direct'):
         message_dict['hazard'] = temp_data['hazard']
     except:
         message_dict['hazard'] = temp_data['rationale']
-    # message_dict['Entity #1'] = temp_data['Entity #1']
-    # message_dict['Entity #2'] = temp_data['Entity #2']
-    # message_dict['Entity #3'] = temp_data['Entity #3']
+    message_dict['Entity #1'] = temp_data['Entity #1']
+    message_dict['Entity #2'] = temp_data['Entity #2']
+    message_dict['Entity #3'] = temp_data['Entity #3']
 
     img_retriever = ImageRetriever(
         root_path=os.path.join(os.environ['ROOT'], ''),