Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
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
|