Update main.py

main.py

@@ -119,6 +119,7 @@ def predict_revenue():
 
     # Ensure 'date' column is datetime
     df['date'] = pd.to_datetime(df['date'])
+    df['date'] = df['date'].dt.tz_localize(None)
 
     # Set 'date' as index
     df = df.sort_values("date").set_index("date")