Update app.py

app.py

@@ -264,11 +264,18 @@ def plot():
 
 
     for x, y in dataset_val.take(n):
-        show_plot(
-            [x[0][:, 1].numpy(), y[0].numpy(), model.predict(x)[0]],
-            12,
-            f"{n} Step Prediction",
-        )
+	if ("0" == "mycustom"):
+        	show_plot(
+	            [x[0][:, 1].numpy(), y[0].numpy(), model.predict(x)[0]],
+        	    12,
+	            f"{n} Step Prediction",
+        	)
+	if ("0" != "mycustom"):
+        	show_plot(
+	            [x[0][:, 0].numpy(), y[0].numpy(), model.predict(x)[0]],
+        	    12,
+	            f"{n} Step Prediction",
+        	)
 
 fig = plot()
 st.pyplot(fig)