Update app.py

app.py

@@ -22,8 +22,9 @@ def load_file(uploaded_file):
 
 # Train the model if it doesn't already exist
 def train_model(data):
-    data['Hour'] = pd.to_datetime(data['Timestamp']).dt.hour
-    X = data[['Hour', 'Temperature', 'CloudCover']]
+   # data['Hour'] = pd.to_datetime(data['Timestamp']).dt.hour
+    data['Hour'] = pd.to_datetime(data['DATE_TIME']).dt.hour
+    X = data[['Hour', 'Temperature', 'CloudCover','DATE_TIME','PLANT_ID','SOURCE_KEY','DC_POWER','AC_POWER','DAILY_YIELD','TOTAL_YIELD']]
     y = data['SolarOutput']
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
     model = LinearRegression()