Update app.py

app.py

@@ -48,8 +48,8 @@ model.load_state_dict(torch.load(model_path))
 
 model.eval()
 
-def predict_and_plot(week1, week2, week3, week4):
-    last_four_weeks = [week1, week2, week3, week4]
+def predict_and_plot(week4, week3, week2, week1):
+    last_four_weeks = [week4, week3, week2, week1]
 
     custom_input = prepare_custom_input(last_four_weeks, seq_length, scaler)
 
@@ -63,9 +63,9 @@ def predict_and_plot(week1, week2, week3, week4):
     predicted_value = f"{predicted_kilometers[0][0]:.2f}"
 
     weeks = ['Week -4', 'Week -3', 'Week -2', 'Week -1', 'Predicted Week']
-    actual_values = [week1, week2, week3, week4, predicted_value]
+    actual_values = [week4, week3, week2, week1, predicted_value]
     
-    plt.figure(figsize=(8, 5))
+    plt.figure(figsize=(10, 6))
     plt.plot(weeks, actual_values, marker='o', label='Total Kilometers')
     plt.axvline(x='Predicted Week', color='red', linestyle='--', label='Predicted Week')
     plt.title('Total Kilometers for Last 4 Weeks and Prediction')