Spaces:
Runtime error
Runtime error
Commit
•
383ddb3
1
Parent(s):
868057f
running with knn model file
Browse files
app.py
CHANGED
|
@@ -13,7 +13,8 @@ from sklearn.feature_extraction.text import CountVectorizer
|
|
| 13 |
import joblib
|
| 14 |
|
| 15 |
# Load the saved model
|
| 16 |
-
model_file = "Drug_Book.joblib"
|
|
|
|
| 17 |
loaded_model = load(model_file)
|
| 18 |
|
| 19 |
def predict_drug(text_input):
|
|
|
|
| 13 |
import joblib
|
| 14 |
|
| 15 |
# Load the saved model
|
| 16 |
+
#model_file = "Drug_Book.joblib"
|
| 17 |
+
model_file = "drug_book_knn.joblib"
|
| 18 |
loaded_model = load(model_file)
|
| 19 |
|
| 20 |
def predict_drug(text_input):
|