diff --git a/viz/renderer.py b/viz/renderer.py
index ed81697..1d2ad6c 100644
--- a/viz/renderer.py
+++ b/viz/renderer.py
@@ -382,5 +382,6 @@ class Renderer:
             img = img.cpu().numpy()
             img = Image.fromarray(img)
         res.image = img
+        res.w = ws.cpu().numpy()
 
 #----------------------------------------------------------------------------