Spaces:
Configuration error
Configuration error
import gradio as gr | |
import pickle | |
import pandas as pd | |
from sklearn.metrics.pairwise import cosine_similarity | |
# Load model and dataset | |
with open("recommender_model.pkl", "rb") as f: | |
model = pickle.load(f) | |
posts_df = pd.read_csv("posts_cleaned.csv") # your full dataset with post content | |
post_embeddings = model["embeddings"] # precomputed post embeddings | |
vectorizer = model["vectorizer"] # for transforming user input | |
# Predict function | |
def recommend_from_input(user_text): | |
user_vec = vectorizer.encode([user_text]) | |
sims = cosine_similarity(user_vec, post_embeddings)[0] | |
top_idxs = sims.argsort()[-5:][::-1] | |
top_posts = posts_df.iloc[top_idxs]["post_text"].tolist() | |
return "\n\n".join(top_posts) | |
# Gradio UI | |
interface = gr.Interface( | |
fn=recommend_from_input, | |
inputs="text", | |
outputs="text", | |
title="AI Content Recommender", | |
description="Enter a sample interest or post to receive recommendations" | |
) | |
interface.launch() | |