Update app.py

app.py

@@ -16,6 +16,9 @@ from detectron2.utils.visualizer import ColorMode, Visualizer
 from math import ceil
 import uuid
 from flask import Flask, request, send_file
+import matplotlib
+matplotlib.use('Agg')
+
 
 app = Flask(__name__)
 
@@ -196,5 +199,5 @@ def predict():
 
     return send_file(submission_det_path, as_attachment=True)
 
-# if __name__ == '__main__':
-#     app.run(host='0.0.0.0', port=8888)
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=8888)