Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -60,35 +60,35 @@ def pre_process(x):
|
|
| 60 |
|
| 61 |
return " ".join(x)
|
| 62 |
|
| 63 |
-
@st.cache_resource
|
| 64 |
-
def load_model():
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
|
| 70 |
-
model, label_encoder = load_model()
|
| 71 |
|
| 72 |
-
def predict_category(text):
|
| 73 |
-
|
| 74 |
|
| 75 |
-
|
| 76 |
|
| 77 |
-
|
| 78 |
-
|
| 79 |
|
| 80 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
|
| 84 |
-
|
| 85 |
|
| 86 |
# Custom CSS
|
| 87 |
st.markdown(
|
| 88 |
"""
|
| 89 |
<style>
|
| 90 |
body {
|
| 91 |
-
background-image: url('https://cdn-uploads.huggingface.co/production/uploads/
|
| 92 |
background-size: cover;
|
| 93 |
background-repeat: no-repeat;
|
| 94 |
background-attachment: fixed;
|
|
|
|
| 60 |
|
| 61 |
return " ".join(x)
|
| 62 |
|
| 63 |
+
# @st.cache_resource
|
| 64 |
+
# def load_model():
|
| 65 |
+
# model = keras.models.load_model("model_m3_new.keras")
|
| 66 |
+
# with open("label_encoder_m5.pkl", 'rb') as file:
|
| 67 |
+
# label_encoder = pickle.load(file)
|
| 68 |
+
# return model, label_encoder
|
| 69 |
|
| 70 |
+
# model, label_encoder = load_model()
|
| 71 |
|
| 72 |
+
# def predict_category(text):
|
| 73 |
+
# cleaned_text = pre_process(text)
|
| 74 |
|
| 75 |
+
# vectorizer = keras.models.load_model("vec_text_m3_new.keras")
|
| 76 |
|
| 77 |
+
# # Vectorizing the pre-processed text
|
| 78 |
+
# text_vectorized = pad_sequences(vectorizer.predict(np.array([cleaned_text])).numpy(), padding='pre', maxlen=128)
|
| 79 |
|
| 80 |
+
# # Model prediction
|
| 81 |
+
# prediction = model.predict(text_vectorized)
|
| 82 |
+
# category_idx = np.argmax(prediction, axis=1)[0]
|
| 83 |
|
| 84 |
+
# return label_encoder.inverse_transform([category_idx])[0], cleaned_text
|
| 85 |
|
| 86 |
# Custom CSS
|
| 87 |
st.markdown(
|
| 88 |
"""
|
| 89 |
<style>
|
| 90 |
body {
|
| 91 |
+
background-image: url('https://cdn-uploads.huggingface.co/production/uploads/675fab3a2d0851e23d23cad3/WwOWG8MBGYxHnIeM2Dowo.webp');
|
| 92 |
background-size: cover;
|
| 93 |
background-repeat: no-repeat;
|
| 94 |
background-attachment: fixed;
|