app.py

from flask import Flask, render_template, url_for, request,redirect
from flask_bootstrap import Bootstrap
import os
import inference

app = Flask(__name__)

Bootstrap(app)
'''
Routes
'''
@app.route('/', methods=['GET', 'POST'])
def index():
    if request.method == 'POST':
        uploaded_file = request.files['file']
        if uploaded_file.filename != '':
            image_path = os.path.json('static', uploaded_file.filename)
            uploaded_file.save(image_path)
            class_name = inference.get_prediction(image_path)
            print('CLASS NAME=', class_name)
            result = {
                'class_name':class_name,
                'image_path': image_path,
            }

            return render_template('show.html', result=result)

    return render_template('index.html')

if __name__ == '__main__':
    app.run(debug = True)