|
@@ -28,6 +28,7 @@ os.environ['CUDA_VISIBLE_DEVICES']='0'
|
|
app = Flask(__name__)
|
|
app = Flask(__name__)
|
|
|
|
|
|
# define a predict function as an endpoint
|
|
# define a predict function as an endpoint
|
|
|
|
+
|
|
@app.route("/process_image", methods=["POST"])
|
|
@app.route("/process_image", methods=["POST"])
|
|
def process_image():
|
|
def process_image():
|
|
source_url = request.json["source_url"]
|
|
source_url = request.json["source_url"]
|
|
@@ -38,7 +39,7 @@ def process_image():
|
|
os.mkdir(upload_directory)
|
|
os.mkdir(upload_directory)
|
|
|
|
|
|
random_filename = str(uuid4()) + '.png'
|
|
random_filename = str(uuid4()) + '.png'
|
|
-
|
|
|
|
|
|
+
|
|
image_colorizer.plot_transformed_image_from_url(url=source_url, path=os.path.join(upload_directory, random_filename), figsize=(20,20),
|
|
image_colorizer.plot_transformed_image_from_url(url=source_url, path=os.path.join(upload_directory, random_filename), figsize=(20,20),
|
|
render_factor=render_factor, display_render_factor=True, compare=False)
|
|
render_factor=render_factor, display_render_factor=True, compare=False)
|
|
|
|
|
|
@@ -49,7 +50,6 @@ def process_image():
|
|
|
|
|
|
return callback
|
|
return callback
|
|
|
|
|
|
-
|
|
|
|
@app.route("/process_video", methods=["POST"])
|
|
@app.route("/process_video", methods=["POST"])
|
|
def process_video():
|
|
def process_video():
|
|
source_url = request.json["source_url"]
|
|
source_url = request.json["source_url"]
|