LinDee commited on
Commit
bf673d6
·
verified ·
1 Parent(s): 133837a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -2,13 +2,16 @@ import gradio as gr
2
  import pickle
3
  import pandas as pd
4
  from sklearn.metrics.pairwise import cosine_similarity
 
5
 
6
  # Load model and dataset
7
  with open("recommender_model.pkl", "rb") as f:
8
  model = pickle.load(f)
9
 
10
  posts_df = pd.read_csv("posts_cleaned.csv") # your full dataset with post content
11
- post_embeddings = model["embeddings"] # precomputed post embeddings
 
 
12
  vectorizer = model["vectorizer"] # for transforming user input
13
 
14
  # Predict function
 
2
  import pickle
3
  import pandas as pd
4
  from sklearn.metrics.pairwise import cosine_similarity
5
+ from sklearn.metrics.pairwise import cosine_similarity
6
 
7
  # Load model and dataset
8
  with open("recommender_model.pkl", "rb") as f:
9
  model = pickle.load(f)
10
 
11
  posts_df = pd.read_csv("posts_cleaned.csv") # your full dataset with post content
12
+ post_texts = posts["post_text"].astype(str).tolist()
13
+ post_embeddings = model.encode(post_texts, convert_to_tensor=False)
14
+ # precomputed post embeddings
15
  vectorizer = model["vectorizer"] # for transforming user input
16
 
17
  # Predict function