Update app.py

app.py

@@ -1,5 +1,5 @@
 from flask import Flask, request, send_file, jsonify
-import cv2, io, random
+import cv2, io, random, os
 import numpy as np
 
 app = Flask(__name__)