Update app.py

app.py

@@ -29,17 +29,17 @@ def main():
                                                  })
             )
             if result[0] == 0:
-                result = "Red "
+                result = "Red Dwarf"
             elif result[0] == 1:
-                result = "Brown"
+                result = "Brown Dwarf"
             elif result[0] == 2:
-                result = "White"
+                result = "White Dwarf"
             elif result[0] == 3:
                 result = "Main Sequence"
             elif result[0] == 4:
-                result = "Main "
+                result = "Super Giants"
             else:
-                result = "Iris-virginica"
+                result = "Hyper Giants"
                 
             st.write(f"Your predicted class is {result}")