Update app.py

app.py

@@ -1,17 +1,26 @@
+import sys
+import os
+
+# Add the path to the project directory where the 'models' folder is located
+sys.path.append(os.path.dirname(os.path.abspath(__file__)))
+
+# Now you can import the model as usual
+from models.neural_network.inference import load_model_and_preprocessor
 
 import gradio as gr
 import pandas as pd
 import numpy as np
 from models.neural_network.inference import load_model_and_preprocessor
 
-
 # Load the pre-trained model and preprocessor
 nn_model, nn_preprocessor = load_model_and_preprocessor('nn_model.keras', 'nn_preprocessor.pkl')
 
 # Load the unique aircraft data and airport distances
 aircraft_data = pd.read_csv('aircraft_data.csv').drop_duplicates(subset='model')
-airport_data = pd.read_csv('airport_distances.csv')
+aircraft_dict = aircraft_data.set_index('model').to_dict(orient='index')
 
+airport_data = pd.read_csv('airport_distances.csv')
+airport_dict = airport_data.set_index(['Origin_Airport', 'Destination_Airport']).to_dict(orient='index')
 
 
 def predict_fuel_burn(model_name, origin, destination, seats, distance):