typo

app.py

@@ -122,7 +122,7 @@ def generate_imgs(x: torch.Tensor,
     out_baseline_pil = to_pil(process_img(out_baseline)[0].to('cpu'))
 
 
-    # Free memory
+    ### Free memory
     del x, y, out, out_baseline, y_plot
     torch.cuda.empty_cache()
     print(f"[After inference] CUDA current allocated: {torch.cuda.memory_allocated() / 1024**2:.2f} MB")