Update app.py

app.py

@@ -35,12 +35,16 @@ if st.button('Predict'):
                              'hours_per_week': hours_per_week, 'new_occupation': occu, 'sex':sex},index=[0])
 
     hasil_log_reg = model_log_reg.predict(data_inf)[0]
+    if hasil_log_reg == 0:
+        hasil_log_reg = 'Dibawah $ 50.000 Per Tahun'
+    else:
+        hasil_log_reg = 'Diatas $ 50.000 Per Tahun'
     st.write(f'Kelas Pendapatan Anda Menurut Model Logistic Regression:     {hasil_log_reg}')
 
     hasil_svm = model_svc.predict(data_inf)[0]
     if hasil_svm == 0:
-        hasil_svm = '<=50K'
+        hasil_svm = 'Dibawah $ 50.000 Per Tahun'
     else:
-        hasil_svm = '>50K'
+        hasil_svm = 'Diatas $ 50.000 Per Tahun'
     st.write(f'Kelas Pendapatan Anda Menurut Model Logistic SVC : {hasil_svm}')