| import joblib | |
| import pandas as pd | |
| import streamlit as st | |
| model = joblib.load('model (3).joblib') | |
| unique_values = joblib.load('unique_values (3).joblib') | |
| unique_CL = unique_values["Color"] | |
| unique_SC = unique_values["Spectral_Class"] | |
| def main(): | |
| st.title("My star type") | |
| st.image("https://img.freepik.com/free-vector/different-types-stars-dark-space_1308-37762.jpg") | |
| st.text("example -> \nK = 27739, L = 849420 , R = 1252, A_M = -7.59 , Color = Blue ,Spectral_Class = B\n -> Type = Hyper Giants") | |
| with st.form("questionaire"): | |
| K = st.number_input('Temperature') | |
| L = st.number_input('Relative Luminosity') | |
| R = st.number_input('Relative Radius') | |
| AM = st.number_input('Absolute Magnitude') | |
| Color = st.selectbox("General Color of Spectrum",options=unique_CL) | |
| Spectral_Class = st.selectbox("Spectral_Class",options=unique_SC) | |
| clicked = st.form_submit_button("Predict type") | |
| if clicked: | |
| result = model.predict(pd.DataFrame({"Temperature(K)": [K], | |
| "Relative Luminosity(Watts)": [L], | |
| "Relative Radius(m)": [R], | |
| "Absolute Magnitude": [AM], | |
| "Color": [Color], | |
| "Spectral_Class": [Spectral_Class] | |
| }) | |
| ) | |
| if result[0] == 0: | |
| result = "Red Dwarf" | |
| elif result[0] == 1: | |
| result = "Brown Dwarf" | |
| elif result[0] == 2: | |
| result = "White Dwarf" | |
| elif result[0] == 3: | |
| result = "Main Sequence" | |
| elif result[0] == 4: | |
| result = "Super Giants" | |
| else: | |
| result = "Hyper Giants" | |
| st.write(f"star type... {result}") | |
| if __name__=="__main__": | |
| main() |