Delete app.py
Browse files
app.py
DELETED
@@ -1,42 +0,0 @@
|
|
1 |
-
import os
|
2 |
-
import pickle
|
3 |
-
import streamlit as st
|
4 |
-
import tensorflow as tf
|
5 |
-
from tensorflow.keras.layers import TextVectorization
|
6 |
-
|
7 |
-
|
8 |
-
@st.cache_resource
|
9 |
-
def load_model():
|
10 |
-
model = tf.keras.models.load_model(os.path.join("model", "toxmodel.keras"))
|
11 |
-
return model
|
12 |
-
|
13 |
-
|
14 |
-
@st.cache_resource
|
15 |
-
def load_vectorizer():
|
16 |
-
from_disk = pickle.load(open(os.path.join("model", "vectorizer.pkl"), "rb"))
|
17 |
-
new_v = TextVectorization.from_config(from_disk['config'])
|
18 |
-
new_v.adapt(tf.data.Dataset.from_tensor_slices(["xyz"])) # fix for Keras bug
|
19 |
-
new_v.set_weights(from_disk['weights'])
|
20 |
-
return new_v
|
21 |
-
|
22 |
-
|
23 |
-
st.title("Toxic Comment Test")
|
24 |
-
st.divider()
|
25 |
-
model = load_model()
|
26 |
-
vectorizer = load_vectorizer()
|
27 |
-
default_prompt = "I love you man, but fuck you!"
|
28 |
-
input_text = st.text_area("Comment:", default_prompt, height=150).lower()
|
29 |
-
if st.button("Test"):
|
30 |
-
if not input_text:
|
31 |
-
st.write("⚠ Warning: Empty prompt.")
|
32 |
-
elif len(input_text) < 15:
|
33 |
-
st.write("⚠ Warning: Model is far less accurate with a small prompt.")
|
34 |
-
if input_text == default_prompt:
|
35 |
-
st.write("Expected results from default prompt are positive for 0 and 2")
|
36 |
-
with st.spinner("Testing..."):
|
37 |
-
inputv = vectorizer([input_text])
|
38 |
-
output = model.predict(inputv)
|
39 |
-
res = (output > 0.5)
|
40 |
-
st.write(["toxic","severe toxic","obscene","threat","insult","identity hate"], res)
|
41 |
-
st.write(output)
|
42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|