Update inference.py

inference.py

@@ -236,6 +236,7 @@ def read_frames():
 		yield frame
 
 def main():
+    out = None
 	if not os.path.isfile(args.face):
 		raise ValueError('--face argument must be a valid path to video/image file')
 
@@ -342,7 +343,8 @@ def main():
 			f[y1:y2, x1:x2] = p
 			out.write(f)
 
-	out.release()
+    if out != None:
+    	out.release()
 
 	command = 'ffmpeg -y -i {} -i {} -strict -2 -q:v 1 {}'.format(args.audio, 'temp/result.avi', args.outfile)
 	subprocess.call(command, shell=platform.system() != 'Windows')