winwithakash commited on
Commit
aea8b2a
1 Parent(s): 36c5415

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import joblib
6
  import datetime as dt
7
 
8
  ## loading in trained model
9
- model = joblib.load('FlightPrice.pkl')
10
 
11
  @st.cache()
12
  def make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline):
@@ -100,7 +100,7 @@ def main():
100
  if predict:
101
  with st.spinner('Wait for prediction....'):
102
  t = make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline)
103
- st.success(f'Fair Price will be around Rs.{t}')
104
 
105
  if __name__=='__main__':
106
  main()
 
6
  import datetime as dt
7
 
8
  ## loading in trained model
9
+ model = joblib.load('GradientBoost_Flight_Fair_Model')
10
 
11
  @st.cache()
12
  def make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline):
 
100
  if predict:
101
  with st.spinner('Wait for prediction....'):
102
  t = make_predictions(journey_date, journey_time, arrival_date, arrival_time, source, destination, stops, airline)
103
+ st.success(f'Fair Price will be around Rs.{t - 1000}')
104
 
105
  if __name__=='__main__':
106
  main()