Update app.py

app.py

@@ -26,12 +26,17 @@ def main():
         st.write("Your uploaded data: ", df.head())
 
     # Create stock data
+    stock_data = None
     if st.button("Create Stock Data"):
         stock_data = torch.rand(128, model.config['ts_size'], model.config['z_dim'])
         stock_data = normalize(stock_data, min_val=model.config['min_val'], max_val=model.config['max_val'])
         st.write("Your stock data has been created: ", stock_data)
+        
     
     if st.button("Get Stock Embedding"):
+        if stock_data == None:
+            stock_data = torch.zeros(128, model.config['ts_size'], model.config['z_dim'])
+            stock_data = normalize(stock_data, min_val=model.config['min_val'], max_val=model.config['max_val'])
         stock_embedding = model.get_embedding(stock_data=stock_data, embedding_used='encoder')
         st.write("Your stock embedding has been created: ", stock_embedding)