Essay-Scoring / app.py
emirkaanozdemr's picture
Upload 3 files
59d5a40 verified
raw
history blame contribute delete
891 Bytes
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)