Update app.py

app.py

@@ -62,8 +62,8 @@ svm_classifier = read_model("https://github.com/manika-lamba/ml/raw/main/model-t
 preprocessing = read_tf("https://github.com/manika-lamba/ml/raw/main/tfidf-topics.pkl")
 
 
-# Function to predict the category for a given abstract
-def predict_category(abstract):
+# Function to predict the topic for a given abstract
+def predict_topic (abstract):
     # Preprocess the abstract
     abstract_preprocessed = preprocessing.transform([abstract])
     # Make prediction
@@ -93,8 +93,8 @@ if uploaded_file is not None:
     df = pd.read_csv(uploaded_file, encoding='latin-1')
     st.dataframe(df)
     
-    # Tag the "Abstract" column with the corresponding categories
-    df['category'] = df['Abstract'].apply(lambda x: predict_category(x)[0])
+    # Tag the "Abstract" column with the corresponding topics
+    df['topic'] = df['Abstract'].apply(lambda x: predict_topic(x)[0])
     st.dataframe(df)
     
     # Convert DataFrame to CSV and encode it properly