update app.py, add choice button for VisionTSpp base and large

app.py

@@ -210,6 +210,7 @@ def visual_ts_with_quantiles(true_data, pred_median, pred_quantiles_list, model_
 
     pred_quantiles_list.insert(len(QUANTILES)//2, pred_median)
     sorted_quantiles = sorted(zip(QUANTILES, pred_quantiles_list), key=lambda x: x[0])
+
     quantile_preds = [item[1] for item in sorted_quantiles if item[0] != 0.5]
     quantile_vals = [item[0] for item in sorted_quantiles if item[0] != 0.5]
     num_bands = len(quantile_preds) // 2
@@ -217,7 +218,9 @@ def visual_ts_with_quantiles(true_data, pred_median, pred_quantiles_list, model_
 
     for i, ax in enumerate(axes):
         ax.plot(true_data[:, i], label='Ground Truth', color='black', linewidth=1.5)
-        pred_range = np.arange(context_len, context_len + pred_len)
+        # pred_range = np.arange(context_len, context_len + pred_len)
+        pred_range = np.arange(context_len-1, context_len + pred_len)
+
         ax.plot(pred_range, pred_median[:, i], label='Prediction (Median)', color='red', linewidth=1.5)
 
         for j in range(num_bands):
@@ -226,7 +229,9 @@ def visual_ts_with_quantiles(true_data, pred_median, pred_quantiles_list, model_
             ax.fill_between(pred_range, lower_quantile_pred, upper_quantile_pred, color=quantile_colors[j], alpha=0.7, label=f'{int(q_low*100)}-{int(q_high*100)}% Quantile')
         
         y_min, y_max = ax.get_ylim()
-        ax.vlines(x=context_len, ymin=y_min, ymax=y_max, colors='gray', linestyles='--', alpha=0.7)
+        # ax.vlines(x=context_len, ymin=y_min, ymax=y_max, colors='gray', linestyles='--', alpha=0.7)
+        ax.vlines(x=context_len-1, ymin=y_min, ymax=y_max, colors='gray', linestyles='--', alpha=0.7)
+        
         ax.set_ylabel(f'Var {i+1}', rotation=0, labelpad=30, ha='right', va='center')
         ax.grid(True, which='both', linestyle='--', linewidth=0.5)
         ax.margins(x=0)