Ejemplo n.º 1
0
from flask import Flask, render_template, request
from search import Search
from naivebayes import NaiveBayes
from imagesearch import ImageSearch
from imagecaption import ImageCaption
from sklearn.model_selection import train_test_split
from zipfile import ZipFile

with ZipFile('pre_processed_data.zip', 'r') as zipObj:
    zipObj.extractall()

search = Search()
search.init()

naive_bayes = NaiveBayes()
naive_bayes.init()

imagesearch = ImageSearch()
imagesearch.init()

image_caption = ImageCaption()
image_caption.init()

app = Flask(__name__)


@app.route("/")
def hello():
    return '<a href="/search">Search</a><br><a href="/classify">Classify</a><br><a href="/image_search">Image Search</a>'