Spaces:
Build error
Build error
File size: 1,415 Bytes
3d2cf36 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from time import sleep
from flask import Flask, render_template, request, send_file
from werkzeug.utils import secure_filename
from werkzeug.datastructures import FileStorage
import cv2
import os
import glob
import inference_flask as util
app = Flask(__name__)
model, transform, device = util.load_model()
@app.route('/')
def r_upload_file():
return render_template('upload.html')
@app.route('/image', methods = ['GET', 'POST'])
def image():
global model, transform, device
for file in glob.glob('./*'):
if file.endswith('.jpg') or file.endswith('.png') or file.endswith('jpeg'):
os.remove(file)
if request.method == 'POST':
f = request.files['file']
f.save(secure_filename(f.filename))
# inference
util.image_inference(model, transform, device, secure_filename(f.filename))
return send_file(secure_filename(f.filename))
@app.route('/video', methods = ['GET', 'POST'])
def video():
global model, transform, device
for file in glob.glob('./*'):
if file.endswith('.mp4') or file.endswith('.avi'):
os.remove(file)
if request.method == 'POST':
f = request.files['file']
f.save(secure_filename(f.filename))
# inference
util.video_inference(model, transform, device, secure_filename(f.filename))
return send_file(secure_filename(f.filename)+'.avi')
if __name__ == '__main__':
app.run(debug = False) |