Update app.py

app.py

@@ -1,3 +1,8 @@
+import os
+# CRITICAL: Fix for Exit Code 139 segmentation fault on Hugging Face CPU
+os.environ["TF_ENABLE_ONEDNN_OPTS"] = "0"
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
+
 import gradio as gr
 import pickle
 import numpy as np
@@ -8,15 +13,13 @@ from transformers import pipeline
 # ==========================================
 # 1. LOAD MODEL A: Toxic Classifier (LSTM)
 # ==========================================
-# Load the old Keras model and its pickle tokenizer
 lstm_model = tf.keras.models.load_model("toxic_classifier_lstm.h5")
 with open("tokenizer.pkl", "rb") as f:
     lstm_tokenizer = pickle.load(f)
 
 def predict_toxicity(text):
-    # Tokenize and pad for the LSTM
     sequences = lstm_tokenizer.texts_to_sequences([text])
-    padded = pad_sequences(sequences, maxlen=100) # adjust maxlen if yours was different
+    padded = pad_sequences(sequences, maxlen=100) 
     prediction = lstm_model.predict(padded)[0][0]
     
     if prediction > 0.5: