Bhushan4829 commited on
Commit
dca5afd
·
1 Parent(s): 3beeebf

Updated final_code with load

Browse files
Files changed (1) hide show
  1. src/final_code.py +3 -3
src/final_code.py CHANGED
@@ -79,10 +79,10 @@ logging.basicConfig(filename='chatbot_error.log', level=logging.ERROR,
79
 
80
  # Load models and other resources outside of the route
81
  # rf_classifier = load('rf_classifier.joblib')
82
- rf_classifier = hf_hub_download(repo_id, 'rf_classifier.joblib')
83
  # tfidf_vectorizer = load('tfidf_vectorizer.joblib')
84
- tfidf_vectorizer = hf_hub_download(repo_id, 'tfidf_vectorizer.joblib')
85
- mlb = hf_hub_download(repo_id, 'mlb.joblib')
86
  # mlb = load('mlb.joblib')
87
  novels_data = pd.read_csv('novels_preprocessed_data.csv')
88
  doc_processor = DocumentProcessor(novels_data,tfidf_vectorizer)
 
79
 
80
  # Load models and other resources outside of the route
81
  # rf_classifier = load('rf_classifier.joblib')
82
+ rf_classifier = load(hf_hub_download(repo_id, 'rf_classifier.joblib'))
83
  # tfidf_vectorizer = load('tfidf_vectorizer.joblib')
84
+ tfidf_vectorizer = load(hf_hub_download(repo_id, 'tfidf_vectorizer.joblib'))
85
+ mlb = load(hf_hub_download(repo_id, 'mlb.joblib'))
86
  # mlb = load('mlb.joblib')
87
  novels_data = pd.read_csv('novels_preprocessed_data.csv')
88
  doc_processor = DocumentProcessor(novels_data,tfidf_vectorizer)