new links to pickle models
Browse files
app.py
CHANGED
@@ -47,8 +47,8 @@ def read_tf(url):
|
|
47 |
preprocessing = pickle.load(f)
|
48 |
return preprocessing
|
49 |
|
50 |
-
svm_classifier = read_model("https://github.com/manika-lamba/ml/raw/main/model2.pkl")
|
51 |
-
preprocessing = read_tf("https://github.com/manika-lamba/ml/raw/main/preprocessing.pkl")
|
52 |
|
53 |
# Function to predict the category for a given abstract
|
54 |
def predict_category(abstract):
|
|
|
47 |
preprocessing = pickle.load(f)
|
48 |
return preprocessing
|
49 |
|
50 |
+
svm_classifier = read_model("https://github.com/manika-lamba/ml/raw/main/category/model2.pkl")
|
51 |
+
preprocessing = read_tf("https://github.com/manika-lamba/ml/raw/main/category/preprocessing.pkl")
|
52 |
|
53 |
# Function to predict the category for a given abstract
|
54 |
def predict_category(abstract):
|