panchadip commited on
Commit
338aae0
·
verified ·
1 Parent(s): e91ee10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import os
7
  # Load models from the "models" folder
8
  models_dir = "models"
9
  # distilbert_model = joblib.load(os.path.join(models_dir, "distilbert_model.joblib"))
10
- bert_topic_model = joblib.load(os.path.join(models_dir, "bert_topic_model.joblib"))
11
  recommendation_model = joblib.load(os.path.join(models_dir, "recommendation_model.joblib"))
12
 
13
  # Streamlit app layout
 
7
  # Load models from the "models" folder
8
  models_dir = "models"
9
  # distilbert_model = joblib.load(os.path.join(models_dir, "distilbert_model.joblib"))
10
+ bert_topic_model = joblib.load(os.path.join(models_dir, "bertopic_model_max_compressed.joblib"))
11
  recommendation_model = joblib.load(os.path.join(models_dir, "recommendation_model.joblib"))
12
 
13
  # Streamlit app layout