Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
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('
|
9 |
-
'
|
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('
|
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('
|
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()
|
|
|
|