Update src/streamlit_app.py

src/streamlit_app.py

@@ -15,7 +15,6 @@ hf_token = os.getenv("HF_TOKEN")
 if hf_token:
     login(hf_token)
 
-
 # ====================== PREPROCESSING (Same as Task 2) ======================
 
 # ====================== LOAD FINE-TUNED MODEL ======================
@@ -53,7 +52,7 @@ if uploaded_file is not None:
             
             # Classify
             predictions = []
-            for text in df['clean_content']:
+            for text in df['content']:
                 if text.strip() == "":
                     predictions.append("Unknown")
                 else: