Update app.py

app.py

@@ -1577,7 +1577,7 @@ def predict_categories(df):
     # 🔥 Filter: Hanya yang bukan 'Positive'
     df = df[df['temuan_kategori'] != 'Positive'].copy()  # ✅ Filter non-Positive
     
-    if 'temuan_kategori' not in df.columns or df.empty:
+    if 'kategori' not in df.columns or df.empty:
         return pd.DataFrame()
     
     start_month = df['created_at'].min().to_period('M')
@@ -1586,7 +1586,7 @@ def predict_categories(df):
     n_months = len(all_months)
     
     results = []
-    for cat, group in df.groupby('temuan_kategori'):
+    for cat, group in df.groupby('kategori'):
         ts_data = (
             group.groupby(group['created_at'].dt.to_period('M'))
             .size()