Update app.py

app.py

@@ -24,19 +24,13 @@ def main():
     if uploaded_file is not None:
         df = pd.read_csv(uploaded_file)
         st.write("Your uploaded data: ", df.head())
-
-    # Create stock data
-    stock_data = None
-    if st.button("Create Stock Data"):
+    
+    if st.button("Get Stock Embedding"):
+        # Create 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'])
+        # Get embedding
         stock_embedding = model.get_embedding(stock_data=stock_data, embedding_used='encoder')
         st.write("Your stock embedding has been created: ", stock_embedding)