minor

app.py

@@ -8,7 +8,7 @@ models = {
     "convae": torch.load("convae.th", map_location="cpu"),
     "deep_convae": torch.load("deep_convae.th", map_location="cpu"),
 }
-def gen(model, seed, nb_iter, nb_samples, width, height):
+def gen(md, model, seed, nb_iter, nb_samples, width, height):
     torch.manual_seed(int(seed))
     bs = 64
     model = models[model]