Update app.py

app.py

@@ -22,9 +22,10 @@ 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
     data['Hour'] = pd.to_datetime(data['Timestamp']).dt.hour
-    X = data[['Hour', 'Temperature', 'CloudCover','DATE_TIME','PLANT_ID','SOURCE_KEY','DC_POWER','AC_POWER','DAILY_YIELD','TOTAL_YIELD']]
+    #data['Hour'] = pd.to_datetime(data['Timestamp']).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()