custom width

Author: avaamini

mitdeeplearning/lab3.py

@@ -5,12 +5,13 @@
 import numpy as np
 import cv2
 
-def play_video(filename):
+def play_video(filename, width=None):
     encoded = base64.b64encode(io.open(filename, 'r+b').read())
+    video_width = 'width="' + str(width) + '"' if width is not None else ''
     embedded = HTML(data='''
-        <video controls>
-            <source src="data:video/mp4;base64,{0}" type="video/mp4" />
-        </video>'''.format(encoded.decode('ascii')))
+        <video controls {0}>
+            <source src="data:video/mp4;base64,{1}" type="video/mp4" />
+        </video>'''.format(video_width, encoded.decode('ascii')))
 
     return embedded