Update app.py

app.py

@@ -69,23 +69,23 @@ def generate_relations(text, entity1, entity2):
     answer_match = answer_label_regex_pattern.search(modelOut)
     if answer_match:
         if answer_match.group(1)=='0':
-            return f"""{entity1} causes  {entity2}"""
+            return f"""'{entity1}' causes  '{entity2}'"""
         elif answer_match.group(1)=='1':
-            return f"""{entity2} causes  {entity1}"""
+            return f"""'{entity2}' causes  '{entity1}'"""
         elif answer_match.group(1)=='2':
-            return f"""{entity1} enables  {entity2}"""
+            return f"""'{entity1}' enables  '{entity2}'"""
         elif answer_match.group(1)=='3':
-            return f"""{entity2} enables  {entity1}"""
+            return f"""'{entity2}' enables  '{entity1}'"""
         elif answer_match.group(1)=='4':
-            return f"""{entity1} prevents  {entity2}"""
+            return f"""'{entity1}' prevents  '{entity2}'"""
         elif answer_match.group(1)=='5':
-            return f"""{entity2} prevents  {entity1}"""
+            return f"""'{entity2}' prevents  '{entity1}'"""
         elif answer_match.group(1)=='6':
-            return f"""{entity1} hinders  {entity2}"""
+            return f"""'{entity1}' hinders  '{entity2}'"""
         elif answer_match.group(1)=='7':
-            return f"""{entity2} hinders  {entity1}"""
+            return f"""'{entity2}' hinders  '{entity1}'"""
         elif answer_match.group(1)=='8':
-            return f"""No causal relation between {entity1} and {entity2}"""
+            return f"""No causal relation between '{entity1}' and '{entity2}'"""
     else:
         return 'No causal relation could be extracted'