Update app.py

app.py

@@ -54,9 +54,6 @@ def train_and_forecast(
         df.sort_index(inplace=True)
     except AttributeError:
         raise gr.Error("Upload a file with the Upload button")
-    print(df)
-    for i in range(prediction_length//2):
-        df.loc[len(df.index)] = [f'predict_{i}', 0.0]
 
     row_offset = offset_calculation(prediction_length, rolling_windows, len(df))
     # prediction_length += int(len(df) * 0.1)
@@ -64,7 +61,7 @@ def train_and_forecast(
         gluon_df = PandasDataset(df, target=df.columns[0])
     except TypeError:
         freq = pd.infer_freq(df.index[:3])
-        date_range = pd.date_range(df.index[0], df.index[-1], freq=freq)
+        date_range = pd.date_range(df.index[0], pd.DateOffset(months=prediction_length//2) + df.index[-1], freq=freq)
         new_df = df.reindex(date_range)
         gluon_df = PandasDataset(new_df, target=new_df.columns[0], freq=freq)