Update app.py

app.py

@@ -61,7 +61,14 @@ def main():
             st.write("classification is:")
             pred = get_prediction(input_string)
 
-            st.write(pred)
+            categories = pred.split(" >> ")
+            formatted_output = []
+            for i, category in enumerate(categories, 1):
+                formatted_output.append(f'Hierarchy {i} classification: {category}')
+
+            for line in formatted_output:
+
+                st.write(line)
 
 if __name__ == "__main__":
     main()