Spaces:
Build error
Build error
import streamlit as st | |
from tensorflow.keras.models import load_model | |
from sklearn.feature_extraction.text import CountVectorizer | |
from textblob import TextBlob | |
from nltk.stem import PorterStemmer | |
import numpy as np | |
pr=PorterStemmer() | |
def lemmafn(text): | |
words=TextBlob(text).words | |
return [pr.stem(word) for word in words] | |
vect=CountVectorizer(stop_words="english",ngram_range=(1,3),max_features=10000) | |
model=load_model("essay_model.h5") | |
st.title("Predict Essayss Scores") | |
essay=st.text_area("Essay") | |
if essay is not None: | |
essay=essay.lower() | |
essay=essay.replace("[^\w\s]","",) | |
essay=essay.replace("\d+","") | |
essay=essay.replace("\n","") | |
essay=[essay] | |
if st.button("Predict"): | |
data=vect.fit_transform(essay) | |
prediction=model.predict(data) | |
predicted_class=np.argmax(prediction) | |
st.write(predicted_class) | |