Update app.py

app.py

@@ -104,6 +104,9 @@ cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')  # Pre-trai
 
 # Define preprocessing functions for classification
 def preprocess_text(text):
+    if not isinstance(text, str):
+        text = ""
+        
     # Step 1: Lowercase the text
     text = text.lower()
 
@@ -161,6 +164,7 @@ with col1:
 
         # Preprocess the content column and predict categories
         if 'content' in df.columns:
+            df['content'] = df['content'].fillna("").astype(str)
             df['preprocessed_content'] = df['content'].apply(preprocess_text)
             df['class'] = df['preprocessed_content'].apply(predict_category)