minor bug

app.py

@@ -14,7 +14,7 @@ def gen(md, model_name, seed, nb_iter, nb_samples, width, height, nb_active, onl
     torch.manual_seed(int(seed))
     bs = 64
     model = models[model_name]
-    if model == "Dense K-Sparse":
+    if model_name == "Dense K-Sparse":
         model.nb_active = nb_active
     samples = iterative_refinement(
         model,