updates view

app.py

@@ -1,7 +1,7 @@
 import numpy as np
 import streamlit as st
 import torch
-import pandas as pd
+import matplotlib.pyplot as plt
 
 import disvae
 import transforms as trans
@@ -30,16 +30,17 @@ def load_decode_function():
     
 decode = load_decode_function()
 
-
-latent_vector = np.array([st.slider(f"L{l}",min_value=-3.0,max_value=3.0,value=0.0) for l in range(3)])
+st.markdown("**Latent Space Parameters**")
+latent_vector = np.array([st.slider(f"Latent Dimension {l}",min_value=-3.0,max_value=3.0,value=0.0) for l in range(3)])
 latent_vector = np.concatenate([latent_vector,np.zeros(7)],axis=0)
 
 ts = decode(latent_vector)
 
-df = pd.DataFrame({
-    "x":np.arange(len(ts)),
-    "y":ts.ravel()
-}
-)
+st.markdown("**Generated Time Series**")
+
+fig, ax = plt.subplots(figsize=(8,4))
+
+ax.plot(ts.ravel())
+ax.set_ylim([0,4])
 
-st.line_chart(df,x="x",y="y")
+st.pyplot(fig)

requirements.txt

@@ -1,5 +1,5 @@
 numpy
-pandas
+matplotlib
 #plotly
 # scipy
 tqdm