Spaces:

manika07
/

topics

Sleeping

manika07 commited on Jan 10, 2024

Commit

a6c8dfb

1 Parent(s): 0221f7c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -58,8 +58,8 @@ def read_tf(url):
         preprocessing = pickle.load(f)
     return preprocessing
-svm_classifier = read_model("https://github.com/manika-lamba/ml/raw/main/category/model2.pkl")
-preprocessing = read_tf("https://github.com/manika-lamba/ml/raw/main/category/preprocessing.pkl")
 # Function to predict the category for a given abstract
 def predict_category(abstract):

         preprocessing = pickle.load(f)
     return preprocessing
+svm_classifier = read_model("https://github.com/manika-lamba/ml/raw/main/category/model-topics.pkl")
+preprocessing = read_tf("https://github.com/manika-lamba/ml/raw/main/category/tfidf-topics.pkl")
 # Function to predict the category for a given abstract
 def predict_category(abstract):