poudel commited on
Commit
0261be6
1 Parent(s): 85b3623

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -8
app.py CHANGED
@@ -5,17 +5,16 @@ import numpy as np
5
  from models.neural_network.inference import load_model_and_preprocessor
6
 
7
  # Load the pre-trained model
8
- nn_model, nn_preprocessor = load_model_and_preprocessor('/Users/ashishpoudel/Downloads/AircraftFuelPrediction-main/saved_models/nn_model.keras',
9
- '/Users/ashishpoudel/Downloads/AircraftFuelPrediction-main/saved_models/nn_preprocessor.pkl')
10
-
11
- xgboost_model = joblib.load('/Users/ashishpoudel/Downloads/AircraftFuelPrediction-main/saved_models/xgboost_model.joblib')
12
 
13
  # Load the unique aircraft data
14
- aircraft_data = pd.read_csv('/Users/ashishpoudel/Downloads/AircraftFuelPrediction-main/datasets/aircraft_data.csv').drop_duplicates(subset='model')
15
  aircraft_dict = aircraft_data.set_index('model').to_dict(orient='index')
16
 
17
  # Load the airport distances data
18
- airport_data = pd.read_csv('/Users/ashishpoudel/Downloads/AircraftFuelPrediction-main/datasets/airport_distances.csv')
19
  airport_dict = airport_data.set_index(['Origin_Airport', 'Destination_Airport']).to_dict(orient='index')
20
 
21
 
@@ -107,5 +106,3 @@ with gr.Blocks() as demo:
107
  submit_btn.click(predict_fuel_burn, inputs=[model_name, origin, destination, seats, distance], outputs=result)
108
 
109
  demo.launch()
110
-
111
-
 
5
  from models.neural_network.inference import load_model_and_preprocessor
6
 
7
  # Load the pre-trained model
8
+ nn_model, nn_preprocessor = load_model_and_preprocessor('saved_models/nn_model.keras',
9
+ 'saved_models/nn_preprocessor.pkl')
10
+ xgboost_model = joblib.load('saved_models/xgboost_model.joblib')
 
11
 
12
  # Load the unique aircraft data
13
+ aircraft_data = pd.read_csv('datasets/aircraft_data.csv').drop_duplicates(subset='model')
14
  aircraft_dict = aircraft_data.set_index('model').to_dict(orient='index')
15
 
16
  # Load the airport distances data
17
+ airport_data = pd.read_csv('datasets/airport_distances.csv')
18
  airport_dict = airport_data.set_index(['Origin_Airport', 'Destination_Airport']).to_dict(orient='index')
19
 
20
 
 
106
  submit_btn.click(predict_fuel_burn, inputs=[model_name, origin, destination, seats, distance], outputs=result)
107
 
108
  demo.launch()