в

app.py

@@ -34,7 +34,7 @@ def compute(dim):
     st.pyplot(w.get_figure())
     
     
-    def random_normal_samples(n, dim=2):
+    def random_normal_samples(n, dim=3):
         return torch.zeros(n, dim).normal_(mean=0, std=1)
     
     samples = np.array(api.model.sample(torch.tensor(random_normal_samples(1000,api.scaled.shape[-1])).float()).detach())