Fixed broken image serving in development server

NVIDIA · Aug 7, 2015 · 0e134b1 · 0e134b1
1 parent f42f473
commit 0e134b1
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/digits/views.py b/digits/views.py
@@ -243,11 +243,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