Update app.py

app.py

@@ -2,22 +2,6 @@ import joblib
 import pandas as pd
 import streamlit as st
 
-CL_DICT = {'Red': 1,
-            'Blue': 2,
-            'White': 3,
-            'Yellow': 4,
-            'Orange': 5
-            }
-            
-SC_DICT = {'M': 1,
-            'B': 2,
-            'O': 3,
-            'A': 4,
-            'F': 5,
-            'K': 6,
-            'G': 7
-            }
-            
 model = joblib.load('model_star.joblib')
 unique_values = joblib.load('unique_values _star.joblib')
 
@@ -40,8 +24,9 @@ def main():
                                                  "Relative Luminosity(Watts)": [L],
                                                  "Relative Radius(m)": [R],
                                                  "Absolute Magnitude": [AM],
-                                                 "Color": [CL_DICT[Color]],
-                                                 "Spectral_Class": [SC_DICT[Spectral_Class]]})
+                                                 "Color": [Color],
+                                                 "Spectral_Class": [Spectral_Class]
+                                                 })
             )
             
             if result[0] == 0: