Update app.py

app.py

@@ -28,8 +28,20 @@ def main():
                                                  "Spectral_Class": [Spectral_Class]
                                                  })
             )
+            if result[0] == 0:
+                result = "Red "
+            elif result[0] == 1:
+                result = "Brown"
+            elif result[0] == 2:
+                result = "White"
+            elif result[0] == 3:
+                result = "Main Sequence"
+            elif result[0] == 4:
+                result = "Main "
+            else:
+                result = "Iris-virginica"
                 
-            st.write("My star type is...", result)
-        
+            st.write(f"Your predicted class is {result}")
+           
 if __name__=="__main__":
     main()