Add application file

app.py

@@ -88,7 +88,7 @@ class TravelDataset(Dataset):
         This method creates a prompt that the model will use to generate a travel plan
         """
         # Format the input text based on available columns
-        destination = row.get('destination', 'Unknown')
+        destination = row.get('dest', 'Unknown')
         days = row.get('days', 3)
         budget = row.get('budget', 'Moderate')
         interests = row.get('interests', 'Culture, Food')
@@ -107,7 +107,7 @@ def load_dataset():
         data = pd.read_csv("hf://datasets/osunlp/TravelPlanner/train.csv")
         
         # Basic data validation
-        required_columns = ['destination', 'days', 'budget', 'interests', 'target']
+        required_columns = ['dest', 'days', 'budget', 'interests', 'target']
         for col in required_columns:
             if col not in data.columns:
                 raise ValueError(f"Missing required column: {col}")