Classification / app.py
MINHCT's picture
Update app.py
4a5caff verified
raw
history blame
No virus
1.17 kB
import joblib # Load Joblib file
import json # Load JSON file
from sklearn.feature_extraction.text import CountVectorizer # Convert text to BOW format
from flask import Flask, request, jsonify # Flask Server
from tensorflow.keras.preprocessing.text import Tokenizer # tokenizing text documents into sequences of tokens (Seq Model)
from tensorflow.keras.preprocessing.sequence import pad_sequences # ensure that all sequences in a dataset have the same length (Seq Model)
from tensorflow.keras.models import load_model # load a pre-trained Keras model saved in the Hierarchical Data Format (HDF5) file format
import numpy as np # scientific computing in Python
import streamlit as st
# load all the models and vectorizer (global vocabulary)
Seq_model = load_model('./LSTM.h5') # Sequential
SVM_Linear_model = joblib.load("./SVM_Linear_Model.joblib") # SVM
logistic_model = joblib.load("./Logistic_Model.joblib") # Logistic
vectorizer = joblib.load('./vectorizer.joblib') # global vocabulary
tokenizer = joblib.load('./tokenizer.joblib')
# Test
x = st.slider('Select a value')
st.write(x, 'squared is', x * x)
if url:
result = categorize(url)
st.json(result)