Update app_star.py

app_star.py

@@ -28,6 +28,38 @@ def main():
     st.title("My star type")
     st.image("https://study.com/academy/lesson/star-formation-main-sequence-dwarf-giant-stars.html")
     st.text('For comparing all models of ML It can be used for prediction')
-    
+    with st.form("questionaire"):
+         K = st.slider('Temperature',min_value=1900,max_value=40000)
+         L = st.slider('Relative Luminosity',min_value=1900,max_value=40000)
+         R = st.slider('Relative Radius',min_value=1900,max_value=40000)
+         AM = st.slider('Absolute Magnitude',min_value=1900,max_value=40000)
+         CL = st.selectbox("General Color of Spectrum",options=unique_CL)
+         SC = st.selectbox("Spectral_Class",options=unique_SC)
+         
+         clicked = st.form_submit_button("Predict type")
+         if clicked:
+            result=model.predict(pd.DataFrame({"Temperature": [K],
+                                               "Relative Luminosity": [L],
+                                               "Relative Radiusducation": [R],
+                                               "Absolute Magnitude": [AM],
+                                               "General Color of Spectrum": [CL],
+                                               "Spectral_Class": [SC]
+                                               }))
+            # Show prediction
+            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"
+            elif result[0] == 5:
+               result = "Hyper Giants"
+               
+            st.success("Your finished income is "+result)
+        
 if __name__=="__main__":
     main()