Update app.py

app.py

@@ -68,23 +68,23 @@ def generate_relations(text, entity1, entity2):
     modelOut = response[len(prompt):].strip()  # remove prompt from output if echoed
     answer_match = answer_label_regex_pattern.search(modelOut)
     if answer_match:
-        if answer_match.group(1)=='0'
+        if answer_match.group(1)=='0':
             return f"""{entity1} causes  {entity2}"""
-        elif answer_match.group(1)=='1'
+        elif answer_match.group(1)=='1':
             return f"""{entity2} causes  {entity1}"""
-        elif answer_match.group(1)=='2'
+        elif answer_match.group(1)=='2':
             return f"""{entity1} enables  {entity2}"""
-        elif answer_match.group(1)=='3'
+        elif answer_match.group(1)=='3':
             return f"""{entity2} enables  {entity1}"""
-        elif answer_match.group(1)=='4'
+        elif answer_match.group(1)=='4':
             return f"""{entity1} prevents  {entity2}"""
-        elif answer_match.group(1)=='5'
+        elif answer_match.group(1)=='5':
             return f"""{entity2} prevents  {entity1}"""
-        elif answer_match.group(1)=='6'
+        elif answer_match.group(1)=='6':
             return f"""{entity1} hinders  {entity2}"""
-        elif answer_match.group(1)=='7'
+        elif answer_match.group(1)=='7':
             return f"""{entity2} hinders  {entity1}"""
-        elif answer_match.group(1)=='8'
+        elif answer_match.group(1)=='8':
             return f"""No causal relation between {entity1} and {entity2}"""
     else:
         return 'No causal relation could be extracted'