Update app.py
Browse files
app.py
CHANGED
@@ -106,9 +106,9 @@ def retrive_data(temp_data, img_key, mode='direct'):
|
|
106 |
message_dict['hazard'] = temp_data['hazard']
|
107 |
except:
|
108 |
message_dict['hazard'] = temp_data['rationale']
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
|
113 |
img_retriever = ImageRetriever(
|
114 |
root_path=os.path.join(os.environ['ROOT'], ''),
|
|
|
106 |
message_dict['hazard'] = temp_data['hazard']
|
107 |
except:
|
108 |
message_dict['hazard'] = temp_data['rationale']
|
109 |
+
message_dict['Entity #1'] = temp_data['Entity #1']
|
110 |
+
message_dict['Entity #2'] = temp_data['Entity #2']
|
111 |
+
message_dict['Entity #3'] = temp_data['Entity #3']
|
112 |
|
113 |
img_retriever = ImageRetriever(
|
114 |
root_path=os.path.join(os.environ['ROOT'], ''),
|