Update app.py

app.py

@@ -66,7 +66,7 @@ def optimize_portfolio(tickers, start_date, end_date, initial_balance):
     while not done:
         action, _ = model.predict(state)
         state, reward, done, _ = env.step(action)
-        portfolio_weights = action / np.sum(action)
+        portfolio_weights = action / (np.sum(action) + 1e-8)
         portfolio_values.append(reward)
 
     plt.figure(figsize=(10, 6))