fix

app.py

@@ -26,8 +26,8 @@ def optimize_model(input_model, sparsity, context, criteria):
     sp = Sparsifier(model, 'filter', context, criteria=eval(criteria))
     sp.sparsify_model(sparsity)
     sp._clean_buffers()
-    pr = Pruner(model, context, criteria=eval(criteria))
-    pr.prune_model(sparsity)
+    pr = Pruner(model, sparsity, context, criteria=eval(criteria))
+    pr.prune_model()
     qu = Quant()
     qu_model = qu.quantize(model)