Fixed broken image serving in development server

NVIDIA · Aug 7, 2015 · c9ce72d · c9ce72d
1 parent cb7ff66
commit c9ce72d
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/digits/views.py b/digits/views.py
@@ -233,11 +233,8 @@ def serve_file(path):
  raise werkzeug.exceptions.NotFound('File not found')
  if os.path.isdir(path):
  raise werkzeug.exceptions.Forbidden('Folder cannot be served')
-
- with open(path, 'r') as infile:
- response = flask.make_response(infile.read())
- response.headers["Content-Disposition"] = "attachment; filename=%s" % os.path.basename(path)
- return response
+
+ return flask.send_file(path)
 
 ### Path Completion