Update app.py

app.py

@@ -12,7 +12,10 @@ def linear_beta_schedule(timesteps):
     beta_start = 0.0001
     beta_end = 0.02
     return torch.linspace(beta_start, beta_end, timesteps)
-    
+
+timesteps= 300
+betas = linear_beta_schedule(timesteps=timesteps)
+
 alphas = 1. - betas
 alphas_cumprod = torch.cumprod(alphas, axis=0)
 alphas_cumprod_prev = F.pad(alphas_cumprod[:-1], (1, 0), value=1.0)