Update app.py

app.py

@@ -118,7 +118,7 @@ def train(symbol: str, p_step: int):
 
     # Create and train a new model per request
     model = Sequential()
-    model.add(LSTM(50, return_sequences=False, input_shape=(time_step, 1)))
+    model.add(LSTM(50, return_sequences=False, input_shape=(time_steps, 1)))
     model.add(Dense(predict_steps))
     model.compile(loss='mean_squared_error', optimizer='adam')
     model.fit(X, y, epochs=epochs, batch_size=32, verbose=1)