Update app.py

app.py

@@ -28,8 +28,7 @@ def forecast():
     train_data = train_data.rename(columns = {'city_x':'city'})
     train_data = train_data.sort_values(by="date", ascending=True).reset_index(drop=True)
     train_data["aqi_next_day"] = train_data.groupby('city')['aqi'].shift(1)
-    
-    train_data.head(5)
+
     X = train_data.drop(columns=["date"]).fillna(0)
     y = X.pop("aqi_next_day")
     X = X.drop(columns =['city','conditions']).fillna(0)