Update inference.py

inference.py

@@ -21,7 +21,12 @@ def lstm_forecast(ts_data, look_back=60, steps=5, epochs=20):
     """
     ts_data: list of historical stock prices
     steps: number of future steps to forecast
+    Automatically adjusts look_back if input is shorter than look_back.
     """
+    # Adjust look_back if input is too short
+    if len(ts_data) < look_back + 1:
+        look_back = max(1, len(ts_data) - 1)
+
     # Normalize
     scaler = MinMaxScaler(feature_range=(0, 1))
     scaled_data = scaler.fit_transform(np.array(ts_data).reshape(-1,1))
@@ -61,7 +66,6 @@ def lstm_forecast(ts_data, look_back=60, steps=5, epochs=20):
     predictions = scaler.inverse_transform(np.array(predictions).reshape(-1,1))
     return predictions.flatten().tolist()
 
-
 def infer(model_type: str, input_data: list, steps: int = 5):
     """
     model_type: 'arima' or 'lstm'